Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

hyp['anchors'] evolution update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-09-03 12:54:22 -07:00
1 parent ffe9eb4238
commit 44cdcc7e0b
2 files changed
+7 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

utils/evolve.sh
+3 -2
View File
@@ -4,11 +4,12 @@
# 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
# t=ultralytics/yolov5:evolve && 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
# python train.py --batch 64 --weights yolov5m.pt --data voc.yaml --img 512 --epochs 50 --evolve --bucket ult/evolve/voc --device $1
python train.py --batch 40 --weights yolov5m.pt --data coco.yaml --img 640 --epochs 30 --evolve --bucket ult/evolve/coco --device $1
done