Update train.py

2019-03-07 13:28:54 +01:00 · 2019-03-07 13:28:54 +01:00 · ff9d343019
commit ff9d343019
parent 473eb8d0c9
1 changed files with 1 additions and 1 deletions
--- 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