diff --git a/utils/evolve.sh b/utils/evolve.sh new file mode 100644 index 000000000000..bd3344042fef --- /dev/null +++ b/utils/evolve.sh @@ -0,0 +1,14 @@ +#!/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:test && 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/voc --device $1 +done