Optuna now continues until the bitter end

Linardos · Sep 5, 2020 · 4162613 · 4162613
1 parent be4b021
commit 4162613
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/torchlib/find_config.py b/torchlib/find_config.py
@@ -192,4 +192,11 @@ def objective(trial: opt.trial):
         vis.show()
 
     else:
-        study.optimize(objective, n_trials=cmdln_args.num_trials)
+        study.optimize(
+            objective,
+            n_trials=cmdln_args.num_trials,
+            catch=(Exception,),
+            gc_after_trial=True,
+            n_jobs=-1,
+        )
+