summaryrefslogtreecommitdiff
path: root/train_sr.py
diff options
context:
space:
mode:
authorHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2018-01-15 04:17:10 +0900
committerHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2018-01-15 04:41:33 +0900
commit7bfc3321e356f24f49c790b578917e8db22bd30d (patch)
tree65e8070b34d22e8b6211bb41e7dd448eb39dccd1 /train_sr.py
parent2be3f03adc5695f82c6ab86da780108f786ed014 (diff)
超解像学習を可能に
Diffstat (limited to 'train_sr.py')
-rw-r--r--train_sr.py1
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(