Former-commit-id: 9b99dba1e6ad5fec942f75418676330f21cef7c2
This commit is contained in:
Arka 2021-10-24 17:14:18 -04:00
parent 6438b1dcdd
commit 2525a30b9d

View file

@ -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():