Skip to content

GitLab

Commit b9f2d427 authored by Myle Ott's avatar Myle Ott Committed by Sergey Edunov
Browse files

Add OOM counter back to logging output

parent 66415206
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
singleprocess_train.py
View file @ b9f2d427
...@@ -177,6 +177,7 @@ def get_training_stats(trainer): ...@@ -177,6 +177,7 @@ def get_training_stats(trainer):
stats['lr'] = trainer.get_lr() stats['lr'] = trainer.get_lr()
stats['gnorm'] = '{:.3f}'.format(trainer.get_meter('gnorm').avg) stats['gnorm'] = '{:.3f}'.format(trainer.get_meter('gnorm').avg)
stats['clip'] = '{:.0%}'.format(trainer.get_meter('clip').avg) stats['clip'] = '{:.0%}'.format(trainer.get_meter('clip').avg)
stats['oom'] = trainer.get_meter('oom').avg
return stats return stats
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment