Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-06-24 15:56:20 +02:00
parent d208f006a1
commit 6b222df35d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -227,7 +227,7 @@ def train(
# Calculate mAP (always test final epoch, skip first 5 if opt.nosave) # Calculate mAP (always test final epoch, skip first 5 if opt.nosave)
if not (opt.notest or (opt.nosave and epoch < 10)) or epoch == epochs - 1: if not (opt.notest or (opt.nosave and epoch < 10)) or epoch == epochs - 1:
with torch.no_grad(): with torch.no_grad():
results, maps = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size_test, model=model, results, maps = test.test(cfg, data_cfg, batch_size=batch_size*2, img_size=img_size_test, model=model,
conf_thres=0.1) conf_thres=0.1)
# Write epoch results # Write epoch results