typo
Former-commit-id: 9b99dba1e6ad5fec942f75418676330f21cef7c2
This commit is contained in:
parent
6438b1dcdd
commit
2525a30b9d
2
train.py
2
train.py
|
@ -112,7 +112,7 @@ def train_net(net,
|
||||||
|
|
||||||
# Evaluation round
|
# Evaluation round
|
||||||
division_step = (n_train // (10 * batch_size))
|
division_step = (n_train // (10 * batch_size))
|
||||||
if division_step > 0:
|
if division_step > 0:
|
||||||
if global_step % division_step == 0:
|
if global_step % division_step == 0:
|
||||||
histograms = {}
|
histograms = {}
|
||||||
for tag, value in net.named_parameters():
|
for tag, value in net.named_parameters():
|
||||||
|
|
Loading…
Reference in a new issue