diff --git a/dataset.py b/dataset.py
index d6f7fe6926cf4e5cf46f35707484b50fced30ca2..ef7da6af497e579d3cdf8a72111e24ac72c68915 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 8792be9e5f2cc85e5d6fc64254edf16596242b96..5294b28309f5a5d56304aa502e8cbf93148da7bd 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