Sheep training flock of 10 fix?

2026-04-24 17:08:47 +01:00
parent e0426bf320
commit d8b4e2c042
1 changed files with 6 additions and 0 deletions
@@ -179,6 +179,12 @@ def main():
        )
        eval_env.close()

+        save_dir = f"runs/smoke_stage{n_sheep}"
+        os.makedirs(save_dir, exist_ok=True)
+        model.save(os.path.join(save_dir, "model"))
+        vn.save(os.path.join(save_dir, "vecnorm.pkl"))
+        print(f"  Model saved to {save_dir}/")
+
        passed = report(n_sheep, success_rate, failure_counts, args.episodes, threshold)
        if not passed:
            all_passed = False