Update on_train_end callback (#7716)
This commit is contained in:
1 parent
1d0a0944c1
commit
aa7a0e91b1
2 files changed
+1
-1
No files matched your search
@@ -466,7 +466,6 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
|
||||
callbacks.run('on_fit_epoch_end', list(mloss) + list(results) + lr, epoch, best_fitness, fi)
|
||||
|
||||
callbacks.run('on_train_end', last, best, plots, epoch, results)
|
||||
LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")
|
||||
|
||||
torch.cuda.empty_cache()
|
||||
return results
|
||||
|
||||
Reference in new issue
Block a user