Tidy up yaml configs (#121)

2024-04-30 16:08:59 +01:00
parent e4e739f4f8
commit 9d60dce6f3
21 changed files with 142 additions and 207 deletions
--- a/examples/2_evaluate_pretrained_policy.py
+++ b/examples/2_evaluate_pretrained_policy.py
@@ -23,8 +23,8 @@ weights_path = folder / "model.pt"
 # Override some config parameters to do with evaluation.
 overrides = [
    f"policy.pretrained_model_path={weights_path}",
-    "eval_episodes=10",
-    "rollout_batch_size=10",
+    "eval.n_episodes=10",
+    "eval.batch_size=10",
    "device=cuda",
 ]