From e9a073e95f487f2f24c7934444d306672c7b420c Mon Sep 17 00:00:00 2001 From: Abul22 Date: Wed, 10 Jul 2019 17:18:57 +1000 Subject: [PATCH] Fix....? from comment : https://github.com/azadis/MC-GAN/issues/15 t_normal_stack = transforms.Compose([ transforms.Normalize((0.5,), (0.5,)), ]) --- data/data_loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/data/data_loader.py b/data/data_loader.py index f0049ee1..21c075de 100644 --- a/data/data_loader.py +++ b/data/data_loader.py @@ -31,8 +31,9 @@ def normalize_stack(input,val=0.5): len_ = input.size(1) mean = (val,)*len_ std = (val,)*len_ - t_normal_stack = transforms.Compose([ - transforms.Normalize(mean,std)]) + t_normal_stack = transforms.Compose([ +transforms.Normalize((0.5,), (0.5,)), +]) return t_normal_stack(input)