summaryrefslogtreecommitdiff
path: root/train_sr.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_sr.py')
-rw-r--r--train_sr.py15
1 files changed, 0 insertions, 15 deletions
diff --git a/train_sr.py b/train_sr.py
index 96f11e7..40f311a 100644
--- a/train_sr.py
+++ b/train_sr.py
@@ -80,20 +80,5 @@ 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(
- y_keys=[
- 'predictor/loss',
- 'predictor/mse',
- 'predictor/adversarial',
- 'discriminator/accuracy',
- 'discriminator/fake',
- 'discriminator/real',
- ],
- x_key='iteration',
- file_name='loss.png',
- trigger=trigger_log,
- ))
-
save_args(arguments, arguments.output)
trainer.run()