diff --git a/ml/train.py b/ml/train.py index 65cdfc1..7228c20 100644 --- a/ml/train.py +++ b/ml/train.py @@ -13,7 +13,7 @@ def parse_args(): parser.add_argument("--device", type=str, default="auto", help="Device to use: 'cpu', 'cuda', or 'auto' to autodetect") parser.add_argument("--use-gui", action="store_true", help="Enable PyBullet GUI during training") parser.add_argument("--num-robots", type=int, default=1, help="Number of robots in the training environment") - parser.add_argument("--robot-spacing", type=float, default=0.5, help="Spacing between robots in meters") + parser.add_argument("--robot-spacing", type=float, default=3.0, 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") return parser.parse_args()