From db08ac54037816a6f7c95d910ddeea6e539ed4fd Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 24 Jul 2019 20:16:35 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- utils/gcp.sh | 6 ++++++ 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 89bfdcea..5f7151e8 100644 --- a/train.py +++ b/train.py @@ -379,8 +379,8 @@ if __name__ == '__main__': if opt.bucket: os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists - for _ in range(1000): # generations to evolve - if os._exists('evolve.txt'): # if evolve.txt exists: select best hyps and mutate + for _ in range(2): # generations to evolve + if os.path.exists('evolve.txt'): # if evolve.txt exists: select best hyps and mutate # Get best hyperparameters x = np.loadtxt('evolve.txt', ndmin=2) x = x[fitness(x).argmax()] # select best fitness hyps diff --git a/utils/gcp.sh b/utils/gcp.sh index 3279b6f8..28b53caa 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -28,6 +28,12 @@ python3 detect.py # Test python3 test.py --save-json +# Evolve +for i in {0..100} +do + python3 train.py --data data/coco.data --img-size 320 --epochs 1 --batch-size 64 --accumulate 1 --evolve --bucket yolov4 +done + # Git pull git pull https://github.com/ultralytics/yolov3 # master git pull https://github.com/ultralytics/yolov3 test # branch