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