This commit is contained in:
Glenn Jocher 2019-04-17 13:34:17 +02:00
parent 582becc4bf
commit b04ea48153
1 changed files with 1 additions and 1 deletions

View File

@ -178,7 +178,7 @@ def train(
print('multi_scale img_size = %g' % dataset.img_size) print('multi_scale img_size = %g' % dataset.img_size)
# Calculate mAP # Calculate mAP
if opt.nosave and epochs < 10: if opt.nosave and epoch < 10:
results = (0, 0, 0, 0, 0) results = (0, 0, 0, 0, 0)
else: else:
with torch.no_grad(): with torch.no_grad():