diff --git a/ppgan/engine/trainer.py b/ppgan/engine/trainer.py index 274ff2847b774054507c101f1fcb0c9af2828ac0..30d2ccda1efead5b486628fc65e7874b33473d1e 100755 --- a/ppgan/engine/trainer.py +++ b/ppgan/engine/trainer.py @@ -98,6 +98,7 @@ class Trainer: # build metrics self.metrics = None + self.is_save_img = True validate_cfg = cfg.get('validate', None) if validate_cfg and 'metrics' in validate_cfg: self.metrics = self.model.setup_metrics(validate_cfg['metrics'])