fixed env eval setup

2026-07-31 18:35:09 +02:00
parent a3e46c3abf
commit 5317ef1299
5 changed files with 105 additions and 72 deletions
@@ -25,7 +25,7 @@ def main():
    parser.add_argument("--seed", type=int, default=0, help="Random seed for reproducibility")
    parser.add_argument("--device", type=str, default="auto", help="Device to use: 'cpu', 'cuda', or 'auto'")
    parser.add_argument("--gui", action="store_true", help="Show the PyBullet GUI during training")
-    parser.add_argument("--num-robots", type=int, default=1, help="Number of robots in the training environment")
+    parser.add_argument("--num-workers", type=int, default=1, help="Number of training environment with one robot each")
    parser.add_argument("--robot-spacing", type=float, default=1.5, help="Spacing between robots in meters")
    parser.add_argument("--start-pose", type=str, choices=["init_deg", "init90_deg"], default="init_deg", help="Initial robot pose at reset")
    args = parser.parse_args()
@@ -37,7 +37,7 @@ def main():
        seed=args.seed,
        device=args.device,
        use_gui=args.gui,
-        num_robots=args.num_robots,
+        num_workers=args.num_workers,
        robot_spacing=args.robot_spacing,
        start_pose=args.start_pose,
    )