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 -2
View File
@@ -37,7 +37,6 @@ def test(data,
plots=True,
log_imgs=0, # number of logged images
compute_loss=None):
# Initialize/load model and set device
training = model is not None
if training: # called by train.py
@@ -227,7 +226,7 @@ def test(data,
print(pf % ('all', seen, nt.sum(), mp, mr, map50, map))
# Print results per class
if (verbose or (nc <= 20 and not training)) and nc > 1 and len(stats):
if (verbose or (nc < 50 and not training)) and nc > 1 and len(stats):
for i, c in enumerate(ap_class):
print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))