From 6adb5ba1f8671d225d019717daabf514b17b31f3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 24 Mar 2021 15:43:32 +0100
Subject: [PATCH] W&B DDP fix 2 (#2587)

Revert unintentional change to test batch sizes caused by PR https://github.com/ultralytics/yolov5/pull/2125

(cherry picked from commit 9f98201dd98651a768acefdd87856c86a031ff89)
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index fb8ada92ff3f..eeeb549a6341 100644
--- a/train.py
+++ b/train.py
@@ -385,7 +385,7 @@ def train(hyp, opt, device, tb_writer=None):
             if not opt.notest or final_epoch:  # Calculate mAP
                 wandb_logger.current_epoch = epoch + 1
                 results, maps, times = test.test(data_dict,
-                                                 batch_size=total_batch_size,
+                                                 batch_size=batch_size * 2,
                                                  imgsz=imgsz_test,
                                                  model=ema.ema,
                                                  single_cls=opt.single_cls,