diff options
Diffstat (limited to 'train_sr.py')
| -rw-r--r-- | train_sr.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/train_sr.py b/train_sr.py index c714aa0..96f11e7 100644 --- a/train_sr.py +++ b/train_sr.py @@ -78,6 +78,7 @@ ext = extensions.snapshot_object(predictor, filename='predictor_{.updater.iterat trainer.extend(ext, trigger=trigger_snapshot) trainer.extend(extensions.LogReport(trigger=trigger_log)) +trainer.extend(extensions.PrintReport(['predictor/loss'])) if extensions.PlotReport.available(): trainer.extend(extensions.PlotReport( |