Updated some parameters, include network architecture searching

RedTachyon · Mar 14, 2022 · 59673de · 59673de
1 parent fdb5182
commit 59673de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/optimize/optuna_crowd.py b/scripts/optimize/optuna_crowd.py
@@ -150,7 +150,7 @@ def objective(trial: optuna.Trial, worker_id: int, path: str) -> float:
     config["model"]["num_actions"] = action_size
 
     wandb.init(
-        project="optuna-sweep",
+        project="optuna-sweep-fixed",
         entity="redtachyon",
         sync_tensorboard=True,
         config=config,

diff --git a/scripts/optimize/run_optuna.sh b/scripts/optimize/run_optuna.sh
@@ -1 +1 @@
-CUDA_VISIBLE_DEVICES=$1 python optuna_crowd.py -e /home/ariel/projects/coltra-rl/builds/crowd-v5/crowd.x86_64 -w $2 -n 25 -o optuna
+CUDA_VISIBLE_DEVICES=$1 python optuna_crowd.py -e /home/ariel/projects/coltra-rl/builds/crowd-v5/crowd.x86_64 -w $2 -n 25 -o egocentric
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		CUDA_VISIBLE_DEVICES=$1 python optuna_crowd.py -e /home/ariel/projects/coltra-rl/builds/crowd-v5/crowd.x86_64 -w $2 -n 25 -o optuna
		CUDA_VISIBLE_DEVICES=$1 python optuna_crowd.py -e /home/ariel/projects/coltra-rl/builds/crowd-v5/crowd.x86_64 -w $2 -n 25 -o egocentric