updates

2019-12-06 19:05:51 -08:00 · 2019-12-06 19:05:51 -08:00 · a066a7b8ea
parent 115b333371
commit a066a7b8ea
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -321,8 +321,8 @@ def train():
        if opt.prebias:
            print_model_biases(model)
        else:
-            # Calculate mAP (always test final epoch, skip first 10 if opt.nosave)
-            if not (opt.notest or (opt.nosave and epoch < 10)) or final_epoch:
+            # Calculate mAP
+            if not opt.notest or final_epoch:
                with torch.no_grad():
                    results, maps = test.test(cfg,
                                              data,