updates

2019-04-17 13:30:28 +02:00
parent 628d7e5081
commit 582becc4bf
1 changed files with 5 additions and 2 deletions
@@ -178,6 +178,9 @@ def train(
                print('multi_scale img_size = %g' % dataset.img_size)

        # Calculate mAP
+        if opt.nosave and epochs < 10:
+            results = (0, 0, 0, 0, 0)
+        else:
            with torch.no_grad():
                results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model, conf_thres=0.1)