forked from ultralytics/yolov5
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathlaunch.txt
23 lines (11 loc) · 1.95 KB
/
launch.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
python -m torch.distributed.launch --nproc_per_node 4 train_kd.py --img 640 --batch 100 --epochs 300 --data coco.yaml --weights='' --cfg='./models/yolov5n.yaml' --device 0,1,2,3
CUDA_VISIBLE_DEVICES=1,2 python -m torch.distributed.run --nproc_per_node 2 train_kd.py --img 640 --batch 160 --epochs 30 --data coco.yaml --weights='yolov5n.pt' --cfg='./models/yolov5n.yaml' --device 1,2
CUDA_VISIBLE_DEVICES=2,3 python -m torch.distributed.run --nproc_per_node 2 train_kd.py --img 640 --batch 128 --epochs 30 --data coco.yaml --weights='yolov5n.pt' --cfg='./models/yolov5n.yaml' --device 2,3 --freeze 10
CUDA_VISIBLE_DEVICES=2,3 python -m torch.distributed.run --nproc_per_node 2 train_kd.py --img 640 --batch 128 --epochs 30 --data coco.yaml --weights='yolov5n.pt' --cfg='./models/yolov5n.yaml' --device 2,3 --freeze 10 --hyp data/hyps/hyp.finetune.yaml
CUDA_VISIBLE_DEVICES=1,2 python -m torch.distributed.run --nproc_per_node 2 train_kd.py --img 640 --batch 144 --epochs 30 --data coco.yaml --weights='yolov5n.pt' --cfg='./models/yolov5n.yaml' --device 1,2 --freeze 10
CUDA_VISIBLE_DEVICES=2 python train_kd_fi.py --img 640 --batch 96 --epochs 80 --data coco.yaml --weights='yolov5n.pt' --cfg='./models/yolov5n.yaml' --teacher_weight yolov5l.pt
CUDA_VISIBLE_DEVICES=1 python train.py --img 640 --batch 80 --epochs 80 --data coco.yaml --cfg='./models/yolov5n.yaml' --weights=''
CUDA_VISIBLE_DEVICES=1 python train_kd_fi.py --img 640 --batch 80 --epochs 80 --data coco.yaml --weights='' --cfg='./models/yolov5n.yaml' --teacher_weight yolov5l.pt
CUDA_VISIBLE_DEVICES=2 python train_kd.py --img 640 --batch 80 --epochs 80 --data coco.yaml --weights='' --cfg='./models/yolov5n.yaml'
CUDA_VISIBLE_DEVICES=1 python train_kd_fi.py --img 640 --batch 80 --epochs 80 --data coco.yaml --weights='' --cfg='./models/yolov5n.yaml' --teacher_weight yolov5m.pt
CUDA_VISIBLE_DEVICES=2 python train_kd.py --img 640 --batch 60 --epochs 80 --data coco.yaml --weights='' --cfg='./models/yolov5n.yaml'