diff options
Diffstat (limited to 'train_sr.py')
| -rw-r--r-- | train_sr.py | 15 |
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() |
