Skip to content

Commit

Permalink
Create evolve.sh for infinite hyp evolution
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Aug 25, 2020
1 parent a8751e5 commit 8666bc5
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions utils/evolve.sh
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 8666bc5

Please sign in to comment.