diff --git a/train.py b/train.py index 5f7151e8..29afa67d 100644 --- a/train.py +++ b/train.py @@ -379,7 +379,7 @@ if __name__ == '__main__': if opt.bucket: os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists - for _ in range(2): # generations to evolve + for _ in range(1): # 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) diff --git a/utils/gcp.sh b/utils/gcp.sh index 28b53caa..eaa843e1 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -29,7 +29,7 @@ python3 detect.py python3 test.py --save-json # Evolve -for i in {0..100} +for i in {0..500} do python3 train.py --data data/coco.data --img-size 320 --epochs 1 --batch-size 64 --accumulate 1 --evolve --bucket yolov4 done