glenn-jocher commited on
Commit
8666bc5
1 Parent(s): a8751e5

Create evolve.sh for infinite hyp evolution

Browse files
Files changed (1) hide show
  1. utils/evolve.sh +14 -0
utils/evolve.sh ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ # Hyperparameter evolution commands (avoids CUDA memory leakage issues)
3
+ # Replaces train.py python generations 'for' loop with a bash 'for' loop
4
+
5
+ # Start on 4-GPU machine
6
+ #for i in 0 1 2 3; do
7
+ # 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
8
+ # sleep 60 # avoid simultaneous evolve.txt read/write
9
+ #done
10
+
11
+ # Hyperparameter evolution commands
12
+ while true; do
13
+ python train.py --batch 64 --weights yolov5m.pt --data voc.yaml --img 512 --epochs 50 --evolve --bucket ult/voc --device $1
14
+ done