summaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
authorHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2018-01-16 16:02:59 +0900
committerHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2018-01-16 16:02:59 +0900
commit86079f0cea1f79beb7cbbec08f6c19191929207a (patch)
tree69ebc560ec11cfba7a0e703f46836b3d7855095a /train.py
parent83608c12e7bb28df1966cbe5b9d86a8e23175044 (diff)
add noise
Diffstat (limited to 'train.py')
-rw-r--r--train.py17
1 files changed, 0 insertions, 17 deletions
diff --git a/train.py b/train.py
index 3e8cced..26490ce 100644
--- a/train.py
+++ b/train.py
@@ -81,22 +81,5 @@ trainer.extend(ext, trigger=trigger_snapshot)
trainer.extend(extensions.LogReport(trigger=trigger_log))
-if extensions.PlotReport.available():
- trainer.extend(extensions.PlotReport(
- y_keys=[
- 'predictor/loss',
- 'predictor/l1',
- 'test/predictor/loss',
- 'train/predictor/loss',
- 'discriminator/accuracy',
- 'discriminator/fake',
- 'discriminator/true',
- 'discriminator/grad',
- ],
- x_key='iteration',
- file_name='loss.png',
- trigger=trigger_log,
- ))
-
save_args(arguments, arguments.output)
trainer.run()