diff --git a/src/mednet/libs/common/models/model.py b/src/mednet/libs/common/models/model.py index cad520975ce784f6183c30b427c0db08ebfaeaab..cc4bf033e2e869956cabd1aeb713ec8dabf5d145 100644 --- a/src/mednet/libs/common/models/model.py +++ b/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