From e5d59918c7d5b7add735a7eb76e9376a153cbb2c Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Fri, 18 Jun 2021 18:36:33 +0800
Subject: [PATCH] fix save img bug (#344)

---
 ppgan/engine/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ppgan/engine/trainer.py b/ppgan/engine/trainer.py
index 274ff28..30d2ccd 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'])
-- 
GitLab