changed post-processing and adam beta values

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):

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