From a066a7b8eaf2f18cc54495d119af57b1aff2f901 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 6 Dec 2019 19:05:51 -0800 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index b7d87f6f..3feb68b9 100644 --- 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,