This commit is contained in:
Glenn Jocher 2019-04-17 17:51:39 +02:00
parent 663e06f4f9
commit 46c55ac3bd
1 changed files with 4 additions and 1 deletions

View File

@ -263,6 +263,10 @@ if __name__ == '__main__':
opt = parser.parse_args()
print(opt, end='\n\n')
if opt.evolve:
opt.notest = True # save time by only testing final epoch
opt.nosave = True # do not save checkpoints
# Train
results = train(
opt.cfg,
@ -279,7 +283,6 @@ if __name__ == '__main__':
# Evolve hyperparameters (optional)
if opt.evolve:
opt.notest = True # save time by only testing final epoch
best_fitness = results[2] # use mAP for fitness
gen = 30 # generations to evolve