Remove number of episodes from the search

RedTachyon · Mar 14, 2022 · cc80792 · cc80792
1 parent 59673de
commit cc80792
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/optimize/optuna_crowd.py b/scripts/optimize/optuna_crowd.py
@@ -40,7 +40,7 @@ class Parser(BaseParser):
 def objective(trial: optuna.Trial, worker_id: int, path: str) -> float:
     # Get some parameters
     lr = trial.suggest_loguniform("lr", 1e-5, 1e-2)
-    n_episodes = trial.suggest_int("n_episodes", 1, 5)
+    n_episodes = 1
 
     steps = n_episodes * 200