Update transformations.py

fixed normalize function
boliaro · Feb 14, 2021 · 82383d8 · 82383d8
1 parent ecb66e4
commit 82383d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/transformations.py b/transformations.py
@@ -142,7 +142,7 @@ def __init__(self,
         self.std = std
 
     def __call__(self, inp, tar):
-        inp = normalize(inp)
+        inp = normalize(inp, mean=self.mean, std=self.std)
 
         return inp, tar
 
@@ -163,4 +163,4 @@ def __call__(self, inp, tar):
         return input_out, target_out
 
     def __repr__(self):
-        return str({self.__class__.__name__: self.__dict__})
+        return str({self.__class__.__name__: self.__dict__})