From 45540c787fd5e19571756d131ac2fde8eb19ac34 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 25 Jun 2019 19:36:11 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 512377d3..bb6f7763 100644 --- a/train.py +++ b/train.py @@ -241,7 +241,7 @@ def train( # 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: with torch.no_grad(): - results, maps = test.test(cfg, data_cfg, batch_size=batch_size * 2, img_size=img_size_test, model=model, + results, maps = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size_test, model=model, conf_thres=0.1) # Write epoch results