amend

facebookresearch · May 3, 2024 · 668c526 · 668c526
1 parent 8b0279a
commit 668c526
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 76 deletions.
diff --git a/fine_tuned/meltingpot/conf/config.yaml b/fine_tuned/meltingpot/conf/config.yaml
@@ -14,8 +14,7 @@ hydra:
 seed: 0
 
 task:
-  max_steps: 1400
-  disable_shooting: True
+  max_steps: 200
 
 model:
   mlp_num_cells: [ 256, 256 ]
@@ -47,7 +46,7 @@ experiment:
   gamma: 0.99
 
   adam_eps: 0.000001
-  lr: 0.00005
+  lr: 0.00025
   clip_grad_norm: True
   clip_grad_val: 5
 

diff --git a/fine_tuned/meltingpot/conf/config_more_perf.yaml b/fine_tuned/meltingpot/conf/config_more_perf.yaml
diff --git a/fine_tuned/meltingpot/meltingpot_run.py b/fine_tuned/meltingpot/meltingpot_run.py
@@ -13,7 +13,7 @@
 from omegaconf import DictConfig, OmegaConf
 
 
-@hydra.main(version_base=None, config_path="conf", config_name="config_more_perf")
+@hydra.main(version_base=None, config_path="conf", config_name="config")
 def hydra_experiment(cfg: DictConfig) -> None:
     hydra_choices = HydraConfig.get().runtime.choices
     task_name = hydra_choices.task