Fix....?

from comment : azadis#15 t_normal_stack = transforms.Compose([ transforms.Normalize((0.5,), (0.5,)), ])
Abul22 · Jul 10, 2019 · e9a073e · e9a073e
1 parent f9fe7cb
commit e9a073e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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)