updates
This commit is contained in:
parent
25bc5e5392
commit
a21b9891b9
2
train.py
2
train.py
|
@ -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'):
|
||||||
|
|
Loading…
Reference in New Issue