updates

2019-04-17 13:34:17 +02:00 · 2019-04-17 13:34:17 +02:00 · b04ea48153
commit b04ea48153
parent 582becc4bf
1 changed files with 1 additions and 1 deletions
--- 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():