summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHiroshiba Kazuyuki <hihokaruta@gmail.com>2017-11-13 22:36:24 +0900
committerHiroshiba Kazuyuki <hihokaruta@gmail.com>2017-11-13 22:36:24 +0900
commit503030684b1eacf4d6c24d47298b84af10871afa (patch)
tree1153c5c1d4a245719dd919dfdd8c1913e13e7996
parente659b95dba4a071ed2669fc2143c5fd644ca44ca (diff)
parent1436960f9dcd9572f7a68b41b9a3ed2de0bbad85 (diff)
Merge branch 'master' of github.com:Hiroshiba/become-yukarin
-rw-r--r--train.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train.py b/train.py
index eeb77d2..b1a2213 100644
--- a/train.py
+++ b/train.py
@@ -58,7 +58,7 @@ trainer.extend(extensions.LogReport(trigger=trigger_log, log_name='log.txt'))
if extensions.PlotReport.available():
trainer.extend(extensions.PlotReport(
- y_keys=['main/loss', 'test/main/loss'],
+ y_keys=['main/loss', 'test/main/loss', 'train/main/loss'],
x_key='iteration',
file_name='loss.png',
trigger=trigger_log,