From bf8f0f3987c15270e51464da2f665b5b2ec53a03 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Aug 2019 20:20:15 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 0937a422..a225d6fe 100644 --- a/train.py +++ b/train.py @@ -39,7 +39,7 @@ def train(): cfg = opt.cfg data = opt.data img_size = opt.img_size - epochs = 3 if opt.prebias else opt.epochs # 500200 batches at bs 16, 117263 images = 273 epochs + epochs = 1 if opt.prebias else opt.epochs # 500200 batches at bs 16, 117263 images = 273 epochs batch_size = opt.batch_size accumulate = opt.accumulate # effective bs = batch_size * accumulate = 16 * 4 = 64 weights = opt.weights # initial training weights