Skip to content

Commit ad102a7

Browse files
yixu.cuiyixu.cui
authored andcommitted
add using train.py script command
1 parent beaf87c commit ad102a7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

run_yolov7.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,8 @@ python -m torch.distributed.launch --nproc_per_node 4 train_aux.py --dataset mot
115115
python train_aux.py --dataset mot20 --workers 8 --device 4 --batch-size 16 --data data/mot20.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights './weights_yolo/yolov7-w6_training.pt' --hyp data/hyp.scratch.custom.yaml --name yolov7-w6-custom-0309-MOT20-train-1gpu-8w-16b-15epochs --epochs 15 2>&1 | tee ./debug/log/info_yolov7-w6-custom-0309-MOT20-train-1gpu-8w-16b-15epochs.log
116116
# MOT20-train_overlap --sync-bn --batch-size 32
117117
python -m torch.distributed.launch --nproc_per_node 2 train_aux.py --dataset mot20 --workers 32 --device 4,5 --batch-size 32 --data data/mot20.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights './weights_yolo/yolov7-w6.pt' --hyp data/hyp.scratch.custom.yaml --name yolov7-w6-custom-0310-MOT20-train-2gpu-32w-32b-15epochs-sync-bn --epochs 15 --sync-bn 2>&1 | tee ./debug/log/info_yolov7-w6-custom-0310-MOT20-train-2gpu-32w-32b-15epochs-sync-bn.log
118+
# MOT20-train_overlap --sync-bn --batch-size 24 --non-aux
119+
python -m torch.distributed.launch --nproc_per_node 4 train.py --dataset mot20 --workers 32 --device 4,5,6,7 --batch-size 24 --data data/mot20.yaml --img 1280 1280 --cfg cfg/training/yolov7.yaml --weights 'weights_yolo/yolov7_training.pt' --hyp data/hyp.scratch.custom.yaml --name yolov7-custom-0408-MOT20-train-4gpu-32w-24b-20epochs-sync-bn --epochs 20 --sync-bn 2>&1 | tee ./debug/log/info_yolov7-custom-0408-MOT20-train-4gpu-32w-24b-20epochs-sync-bn.log
118120

119121

120122
# test

0 commit comments

Comments
 (0)