From ff9d34301934b17f0c5b3589a684068b3db15bef Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 7 Mar 2019 13:28:54 +0100 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3fd6972f..a0dd3222 100644 --- a/train.py +++ b/train.py @@ -87,7 +87,7 @@ def train( t0 = time.time() model_info(model) - n_burnin = min(round(dataloader.nB / 5), 1000) # number of burn-in batches + n_burnin = min(round(dataloader.nB / 5 + 1), 1000) # number of burn-in batches for epoch in range(epochs): epoch += start_epoch