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