This commit is contained in:
Glenn Jocher 2019-06-12 11:37:25 +02:00
parent d5e2daf79d
commit b0b6554eee
1 changed files with 1 additions and 1 deletions

View File

@ -253,7 +253,7 @@ def train(
if not (opt.notest or (opt.nosave and epoch < 10)) or epoch == epochs - 1:
with torch.no_grad():
results, maps = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model,
conf_thres=0.1)
conf_thres=0.001)
# Write epoch results
with open('results.txt', 'a') as file: