updates
This commit is contained in:
parent
582becc4bf
commit
b04ea48153
2
train.py
2
train.py
|
@ -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():
|
||||||
|
|
Loading…
Reference in New Issue