-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathruntrain.sh
executable file
·10 lines (9 loc) · 2.49 KB
/
runtrain.sh
1
2
3
4
5
6
7
8
9
10
# python -m torch.distributed.launch --nproc_per_node 2 --master_port 9528 train.py --resume runs/train/yololit_320_6x_new2/weights/last.pt
# python train.py --data data/coco8.yaml --cfg runs/train/yololit_320_6x_new4/cfg.yaml --hyp data/hyp.scratch.p5_requant.yaml --name continue_test --batch-size 16 --epochs 1 --device 0
# python -m torch.distributed.launch --nproc_per_node 2 --master_port 9532 train.py --cfg cfg/training/yololit_320_4w4a.yaml --weights runs/train/yololit_320_6x_new_continue_0epoch_3600/weights/best.pt --hyp data/hyp.scratch.p5_requant4.yaml --name yololit_320_4w4a_requant4_adam_ema --adam --fresh_optimizer --batch-size 32 --epochs 1800 --device 0,1 --sync-bn
# python -m torch.distributed.launch --nproc_per_node 2 --master_port 9532 train.py --cfg cfg/training/yololit_320_4w4a.yaml --weights runs/train/yololit_320_4w4a_requant4_adam/weights/best.pt --hyp data/hyp.scratch.p5_requant4.yaml --name yololit_320_4w4a_requant4_adam_uadetrac_nc4 --data data/ua-detrac.yaml --freeze 17 --adam --fresh_optimizer --batch-size 32 --epochs 1800 --device 0,1 --sync-bn
# python -m torch.distributed.launch --nproc_per_node 2 --master_port 9532 train.py --cfg cfg/training/quantyolov8_4w4a.yaml --weights runs/train/v8_float_3202/weights/best.pt --hyp data/hyp_v8_finetune.yaml --name quantyolov8_4w4a_independent_acts --img-size 320 320 --batch-size 32 --epochs 600 --device 0,1 --sync-bn
# python -m torch.distributed.launch --nproc_per_node 2 --master_port 9533 train.py --cfg cfg/training/quantyolov8_4w4a_common_act.yaml --weights runs/train/v8_float_3202/weights/best.pt --hyp data/hyp_v8_finetune.yaml --name quantyolov8_4w4a_a4_comact_fropt_frbn --img-size 320 320 --batch-size 32 --epochs 900 --device 0,1 --sync-bn --freeze_bn_stats --fresh_optimizer --fresh_ema --adam
# python train.py --cfg cfg/training/quantyolov8_8w8a.yaml --weights runs/train/v8_float_320/weights/best.pt --hyp data/hyp_v8_finetune_lre3.yaml --name quantyolov8_8w8a --img-size 320 320 --fresh_ema --batch-size 32 --epochs 900 --device 0
# python train.py --cfg cfg/training/quantyolov8_4w4a.yaml --weights runs/train/v8_float_320/weights/best.pt --hyp data/hyp_v8_finetune_lre3.yaml --name quantyolov8_4w4a --img-size 320 320 --fresh_ema --batch-size 32 --epochs 900 --device 0
python train.py --cfg cfg/training/quantyolov8_4w4a.yaml --weights runs/train/v8_float_320/weights/best.pt --hyp data/hyp_v8_finetune_lre3.yaml --name quantyolov8_4w4a_freezebn --img-size 320 320 --fresh_ema --batch-size 32 --epochs 900 --device 0 --freeze_bn_stats