From b04ea48153b86acec8785690fe7192ad848f62a5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 17 Apr 2019 13:34:17 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5ea0aa16..99f3f840 100644 --- a/train.py +++ b/train.py @@ -178,7 +178,7 @@ def train( print('multi_scale img_size = %g' % dataset.img_size) # Calculate mAP - if opt.nosave and epochs < 10: + if opt.nosave and epoch < 10: results = (0, 0, 0, 0, 0) else: with torch.no_grad():