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

singleprocess_train.py

+1 −0
Changes for singleprocess_train.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -177,6 +177,7 @@ def get_training_stats(trainer): 
    stats['lr'] = trainer.get_lr()

    stats['gnorm'] = '{:.3f}'.format(trainer.get_meter('gnorm').avg)

    stats['clip'] = '{:.0%}'.format(trainer.get_meter('clip').avg)

    stats['oom'] = trainer.get_meter('oom').avg

    return stats