max normalization allows to define norm values

mu_map/dataset/normalization.py

@@ -10,9 +10,14 @@ def norm_max(tensor: Tensor) -> Tensor:
 
 
 class MaxNormTransform(Transform):
+    def __init__(self, max_vals: Tuple[float, float] = None):
+        self.max_vals = max_vals
+
     def __call__(
         self, inputs: Tensor, outputs_expected: Tensor
     ) -> Tuple[Tensor, Tensor]:
+        if self.max_vals:
+            return inputs / self.max_vals[0], outputs_expected / self.max_vals[1]
         return norm_max(inputs), outputs_expected