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