Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -379,7 +379,7 @@ if __name__ == '__main__':
if opt.bucket: if opt.bucket:
os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists 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 if os.path.exists('evolve.txt'): # if evolve.txt exists: select best hyps and mutate
# Get best hyperparameters # Get best hyperparameters
x = np.loadtxt('evolve.txt', ndmin=2) x = np.loadtxt('evolve.txt', ndmin=2)

2
utils/gcp.sh
View File

@ -29,7 +29,7 @@ python3 detect.py
python3 test.py --save-json python3 test.py --save-json
# Evolve # Evolve
for i in {0..100} for i in {0..500}
do do
python3 train.py --data data/coco.data --img-size 320 --epochs 1 --batch-size 64 --accumulate 1 --evolve --bucket yolov4 python3 train.py --data data/coco.data --img-size 320 --epochs 1 --batch-size 64 --accumulate 1 --evolve --bucket yolov4
done done