Create evolve.sh for infinite hyp evolution
This commit is contained in:
parent
a8751e50de
commit
8666bc507e
|
|
@ -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
|
||||||
Loading…
Reference in New Issue