Files
yolov26_3d/train_mono3d_single_node.sh
2026-06-24 09:35:46 +08:00

10 lines
431 B
Bash
Executable File

python -m torch.distributed.run --master_port 29501 --nproc_per_node 8 \
train_mono3d.py --device 0,1,2,3,4,5,6,7 \
--epochs 100 --lr0 0.001 --batch 128 --imgsz 768,352 \
--roi roi0 \
--pretrained runs/detect/train_mono3d_roi0_202603251430/weights/best.pt --roi_metrics_only \
--warmup_epochs 10 \
--loss_3d_warmup_epochs 10 \
--loss_3d_ramp_epochs 20 \
--loss_3d_weight_max 0.1 \
--exp_dir=./runs/detect/train_mono3d_roi0_20260327