单目3D初始代码
This commit is contained in:
10
train_mono3d_single_node.sh
Executable file
10
train_mono3d_single_node.sh
Executable file
@@ -0,0 +1,10 @@
|
||||
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
|
||||
Reference in New Issue
Block a user