|
|
@@ -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 |