Remove loss_name value and save loss adaptively

Save it if it is a string, otherwise ignore

tsimcne/tsimcne.py

@@ -36,7 +36,6 @@ def __init__(
         self,
         model=None,
         loss="infonce",
-        loss_name=None,
         metric=None,
         temperature=0.5,
         backbone="resnet18_sm_kernel",
@@ -63,16 +62,15 @@ def __init__(
     ):
         super().__init__()
         ignore_list = [
-            "loss",
             "backbone",
             "projection_head",
             "eval_function",
             "model",
         ]
+        ignore_list += ["loss"] if not isinstance(loss, str) else []
         self.save_hyperparameters(ignore=ignore_list)
         self.model = model
         self.loss = loss
-        self.loss_name = self.loss if loss_name is None else loss_name
         self.metric = metric
         self.temperature = temperature
         self.backbone = backbone