From a21b9891b920fea6920b48c73b7ce8f1fb068e89 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 8 Aug 2019 21:16:09 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 729fd25d..073b4f8d 100644 --- a/train.py +++ b/train.py @@ -229,7 +229,7 @@ def train(cfg, plot_images(imgs=imgs, targets=targets, paths=paths, fname='train_batch%g.jpg' % i) # Hyperparameter burn-in - # n_burn = min(nb // 5 + 1, 1000) # number of burn-in batches + # n_burn = nb - 1 # min(nb // 5 + 1, 1000) # number of burn-in batches # if ni <= n_burn: # for m in model.named_modules(): # if m[0].endswith('BatchNorm2d'):