From 7807337c5f6cc16a1478f65cdbb2185a73ff2baa Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 31 May 2019 14:30:27 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 4b9d828a..77a5d434 100644 --- a/train.py +++ b/train.py @@ -61,7 +61,7 @@ def train( data_cfg, img_size=416, resume=False, - epochs=68, # 500200 batches at bs 4, dataset length 117263 + epochs=100, # 500200 batches at bs 4, 117263 images = 68 epochs batch_size=16, accumulate=4, # effective bs = 64 = batch_size * accumulate multi_scale=False,