From 32f5ea955b6c279c1128ad2c523067b3200be9af Mon Sep 17 00:00:00 2001
From: glenn-jocher <glenn.jocher@ultralytics.com>
Date: Sun, 30 Jun 2019 17:47:10 +0200
Subject: [PATCH] updates

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 790fc20e..6ac50a54 100644
--- a/train.py
+++ b/train.py
@@ -47,9 +47,9 @@ def train(
         data_cfg,
         img_size=416,
         resume=False,
-        epochs=100,  # 500200 batches at bs 4, 117263 images = 68 epochs
-        batch_size=16,
-        accumulate=4,  # effective bs = 64 = batch_size * accumulate
+        epochs=100,  # 500200 batches at bs 16, 117263 images = 273 epochs
+        batch_size=8,
+        accumulate=8,  # effective bs = batch_size * accumulate = 8 * 8 = 64
         freeze_backbone=False,
         transfer=False  # Transfer learning (train only YOLO layers)
 ):