[libs.common.models.model] Do not assume pos_weight is a scalar

src/mednet/libs/common/models/model.py

@@ -228,7 +228,7 @@ class Model(pl.LightningModule):
             self._train_loss_arguments["pos_weight"] = train_weights
             logger.info(
                 f"Balanced training loss {self._loss_type}: "
-                f"`pos_weight={train_weights.item():.3f}`."
+                f"`pos_weight={train_weights}`."
             )
 
             if "validation" in datamodule.val_dataloader().keys():
@@ -245,7 +245,7 @@ class Model(pl.LightningModule):
             self._validation_loss_arguments["pos_weight"] = validation_weights
             logger.info(
                 f"Balanced validation loss {self._loss_type}: "
-                f"`pos_weight={validation_weights.item():.3f}`."
+                f"`pos_weight={validation_weights}`."
             )
 
         # re-instantiates losses for train and validation