verbose on final_epoch (#1997)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-20 18:27:38 -08:00
1 parent d9212140b3
commit 046c37e465
2 files changed
+2 -2

No files matched your search

+1
View File
@@ -344,6 +344,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
single_cls=opt.single_cls,
dataloader=testloader,
save_dir=save_dir,
verbose=nc < 50 and final_epoch,
plots=plots and final_epoch,
log_imgs=opt.log_imgs if wandb else 0,
compute_loss=compute_loss)