This commit is contained in:
Glenn Jocher 2019-07-24 20:32:24 +02:00
parent db08ac5403
commit 169d117870
2 changed files with 2 additions and 2 deletions

View File

@ -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)

View File

@ -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