From 86079f0cea1f79beb7cbbec08f6c19191929207a Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Tue, 16 Jan 2018 16:02:59 +0900 Subject: add noise --- train.py | 17 ----------------- 1 file changed, 17 deletions(-) (limited to 'train.py') 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() -- cgit v1.2.3-70-g09d2