File size: 747 Bytes
8666bc5 44cdcc7 8666bc5 44cdcc7 8666bc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
#!/bin/bash
# Hyperparameter evolution commands (avoids CUDA memory leakage issues)
# Replaces train.py python generations 'for' loop with a bash 'for' loop
# Start on 4-GPU machine
#for i in 0 1 2 3; do
# t=ultralytics/yolov5:evolve && sudo docker pull $t && sudo docker run -d --ipc=host --gpus all -v "$(pwd)"/VOC:/usr/src/VOC $t bash utils/evolve.sh $i
# sleep 60 # avoid simultaneous evolve.txt read/write
#done
# Hyperparameter evolution commands
while true; do
# python train.py --batch 64 --weights yolov5m.pt --data voc.yaml --img 512 --epochs 50 --evolve --bucket ult/evolve/voc --device $1
python train.py --batch 40 --weights yolov5m.pt --data coco.yaml --img 640 --epochs 30 --evolve --bucket ult/evolve/coco --device $1
done
|