From b0b6554eee0477c82158b3ec2572b375b8fdce58 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 12 Jun 2019 11:37:25 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5ba11c57..dddd2980 100644 --- a/train.py +++ b/train.py @@ -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: