Commit
·
8666bc5
1
Parent(s):
a8751e5
Create evolve.sh for infinite hyp evolution
Browse files- 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
|