Browse Source

update

tags/v0.1.0
FengZiYjun 7 years ago
parent
commit
0430067faf
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/trainer.py

+ 1
- 1
fastNLP/core/trainer.py View File

@@ -164,7 +164,7 @@ class BaseTrainer(object):
self.grad_backward(loss)
self.update()

if step % kwargs["n_print"] == 0:
if kwargs["n_print"] > 0 and step % kwargs["n_print"] == 0:
end = time.time()
diff = timedelta(seconds=round(end - kwargs["start"]))
print_output = "[epoch: {:>3} step: {:>4}] train loss: {:>4.2} time: {}".format(


Loading…
Cancel
Save