From 54e43b9ad6a74db8cba8da438c6ae4db3798c63f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 5 Apr 2019 16:19:51 +0200
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 5e527e9f..83eb7ccc 100644
--- a/train.py
+++ b/train.py
@@ -15,7 +15,7 @@ def train(
         data_cfg,
         img_size=416,
         resume=False,
-        epochs=270,
+        epochs=273,  # 500200 batches at bs 64, dataset length 117263
         batch_size=16,
         accumulate=1,
         multi_scale=False,
@@ -75,7 +75,7 @@ def train(
         else:
             cutoff = load_darknet_weights(model, weights + 'darknet53.conv.74')
 
-    # Set scheduler (reduce lr at epoch 250)
+    # Set scheduler (reduce lr at epochs 218, 245, i.e. batches 400k, 450k)
     scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=[218, 245], gamma=0.1,
                                                      last_epoch=start_epoch - 1)