Sheep training flock _ improver

2026-04-25 18:46:41 +01:00
parent 5005128c07
commit 7bfb7d3aae
2 changed files with 18 additions and 8 deletions
@@ -44,7 +44,7 @@ def main():
                   help="Train with n_sheep randomized per episode (no curriculum). "
                        "Total train steps = steps-per-stage * max_sheep.")
    p.add_argument("--n-envs", type=int, default=8)
-    p.add_argument("--max-steps", type=int, default=1500)
+    p.add_argument("--max-steps", type=int, default=2500)
    p.add_argument("--eval-episodes", type=int, default=30)
    p.add_argument("--run-dir", type=str, default=None)
    args = p.parse_args()