This commit is contained in:
Glenn Jocher 2019-08-08 21:16:09 +02:00
parent 25bc5e5392
commit a21b9891b9
1 changed files with 1 additions and 1 deletions

View File

@ -229,7 +229,7 @@ def train(cfg,
plot_images(imgs=imgs, targets=targets, paths=paths, fname='train_batch%g.jpg' % i) plot_images(imgs=imgs, targets=targets, paths=paths, fname='train_batch%g.jpg' % i)
# Hyperparameter burn-in # 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: # if ni <= n_burn:
# for m in model.named_modules(): # for m in model.named_modules():
# if m[0].endswith('BatchNorm2d'): # if m[0].endswith('BatchNorm2d'):