Reset gnorm after each epoch

97a6b139 · Sergey Edunov · Myle Ott · ef43da72 · 97a6b139
Commit 97a6b139 authored Aug 09, 2018 by Sergey Edunov Committed by Myle Ott Sep 03, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

No files found.
--- a/train.py
+++ b/train.py
@@ -155,7 +155,7 @@ def train(args, trainer, task, epoch_itr):
    progress.print(stats)

    # reset training meters
-    for k in ['train_loss', 'train_nll_loss', 'wps', 'ups', 'wpb', 'bsz', 'clip']:
+    for k in ['train_loss', 'train_nll_loss', 'wps', 'ups', 'wpb', 'bsz', 'clip', 'gnorm']:
        meter = trainer.get_meter(k)
        if meter is not None:
            meter.reset()