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-21 05:06:58 +02:00
parent 9f7e71acab
commit c9ed072cc7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -37,7 +37,7 @@ hyp = {'giou': 1.153, # giou loss gain
'lr0': 0.001357, # initial learning rate
'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf)
'momentum': 0.916, # SGD momentum
'weight_decay': 0.000572, # optimizer weight decay
'weight_decay': 0.0000572, # optimizer weight decay
'hsv_s': 0.5, # image HSV-Saturation augmentation (fraction)
'hsv_v': 0.5, # image HSV-Value augmentation (fraction)
'degrees': 5, # image rotation (+/- deg)
@ -328,7 +328,7 @@ def print_mutation(hyp, results):
with open('evolve.txt', 'a') as f: # append result
f.write(c + b + '\n')
x = np.unique(np.loadtxt('evolve.txt', ndmin=2), axis=0) # load unique rows
np.savetxt('evolve2.txt', x[np.argsort(-fitness(x))] , '%11.3g') # save sort by fitness
np.savetxt('evolve.txt', x[np.argsort(-fitness(x))] , '%11.3g') # save sort by fitness
os.system('gsutil cp evolve.txt gs://%s' % opt.bucket) # upload evolve.txt
else:
with open('evolve.txt', 'a') as f: