convert some string equals to in checks for more flexibility

mu_map/dataset/normalization.py

@@ -124,11 +124,11 @@ def norm_by_str(norm: Optional[str]) -> Optional[NormTransform]:
         return None
 
     norm = norm.lower()
-    if norm == "mean":
+    if "mean" in norm:
         return MeanNormTransform()
-    elif norm == "max":
+    elif "max" in norm:
         return MaxNormTransform()
-    elif norm == "gaussian":
+    elif "gaussian" in norm:
         return GaussianNormTransform()
 
     raise ValueError(f"Unknown normalization {norm}")

mu_map/training/loss.py

@@ -28,11 +28,11 @@ def loss_by_string(loss_str: str) -> nn.Module:
     :returns: an executable loss function
     """
     loss_str = loss_str.lower()
-    if loss_str == "l1":
+    if "l1" in loss_str:
         return nn.L1Loss(reduction="mean")
-    elif loss_str == "l2" or loss_str == "mse":
+    elif "l2" in loss_str or "mse" in loss_str:
         return nn.MSELoss(reduction="mean")
-    elif loss_str == "gdl":
+    elif "gdl" in loss_str or "gradientdifferenceloss" in loss_str:
         return GradientDifferenceLoss()
     else:
         raise ValueError(f"Unknown loss function: {loss_str}")