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

hyp evolution bug fix #1160

This commit is contained in:
Glenn Jocher 2020-05-12 09:53:13 -07:00
parent 9f04e175f6
commit 0cf88f046d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -54,7 +54,7 @@ if hyp['fl_gamma']:
print('Using FocalLoss(gamma=%g)' % hyp['fl_gamma']) print('Using FocalLoss(gamma=%g)' % hyp['fl_gamma'])
def train(): def train(hyp):
cfg = opt.cfg cfg = opt.cfg
data = opt.data data = opt.data
epochs = opt.epochs # 500200 batches at bs 64, 117263 images = 273 epochs epochs = opt.epochs # 500200 batches at bs 64, 117263 images = 273 epochs
@ -409,7 +409,7 @@ if __name__ == '__main__':
if not opt.evolve: # Train normally if not opt.evolve: # Train normally
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/') print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
tb_writer = SummaryWriter(comment=opt.name) tb_writer = SummaryWriter(comment=opt.name)
train() # train normally train(hyp) # train normally
else: # Evolve hyperparameters (optional) else: # Evolve hyperparameters (optional)
opt.notest, opt.nosave = True, True # only test/save final epoch opt.notest, opt.nosave = True, True # only test/save final epoch
@ -455,7 +455,7 @@ if __name__ == '__main__':
hyp[k] = np.clip(hyp[k], v[0], v[1]) hyp[k] = np.clip(hyp[k], v[0], v[1])
# Train mutation # Train mutation
results = train() results = train(hyp.copy())
# Write mutation results # Write mutation results
print_mutation(hyp, results, opt.bucket) print_mutation(hyp, results, opt.bucket)