From 9ba2b2a2a37823cbd7e0a1b93d302a0b75e6fd37 Mon Sep 17 00:00:00 2001
From: zeroact <lhj5682@naver.com>
Date: Sat, 22 May 2021 20:01:20 +0900
Subject: [PATCH] changed post-processing and adam beta values

---
 dataset.py | 4 ++--
 train.py   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/dataset.py b/dataset.py
index d6f7fe6..ef7da6a 100644
--- a/dataset.py
+++ b/dataset.py
@@ -23,8 +23,8 @@ def preprocess_image(img, target_shape: tuple):
     return img
 
 def postprocess_image(img):
-    # img = img * 255
-    img = (img - img.min()) / (img.max() - img.min()) * 255
+    img = img * 255
+    img = np.clip(img, 0, 255)
     return img.astype(np.uint8)
 
 class CustomDataset(Dataset):
diff --git a/train.py b/train.py
index 8792be9..5294b28 100644
--- a/train.py
+++ b/train.py
@@ -104,7 +104,7 @@ if __name__ == "__main__":
     print(f"Training start from epoch {initial_epoch}")
     
     # Train Setting
-    model_optim = Adam(model.parameters(), 0.0001)
+    model_optim = Adam(model.parameters(), 0.0001, (0.5, 0.9))
     discrim_optim = Adam(model.discrim.parameters(), 0.0004)
     
     ### Train
-- 
GitLab