fix train.py

chainer · Hakuyume · Jun 2, 2017 · May 30, 2017 · May 30, 2017 · May 30, 2017
commit ee4bd80b80cdaae416dc86f0f2a5e6591b76c923
diff --git a/examples/segnet/train.py b/examples/segnet/train.py
@@ -98,7 +98,7 @@ def transform(in_data):
         trigger=end_trigger))
     trainer.extend(extensions.PrintReport(
         ['epoch', 'iteration', 'elapsed_time', 'lr',
-         'main/loss']),
+         'main/loss', 'validation/main/loss']),
         trigger=log_trigger)
     trainer.extend(extensions.ProgressBar(update_interval=10))