Antonio Mirarchi

[WIP] Introduce additional_labels

1

This PR proposes several enhancements to enable the usage of extra arguments in both training and inference. The key modifications include - [x] standardizing the naming conventions for all extra...

fix lightning hparams conflict

6

DO NOT MERGE YET! In PyTorch Lightning, the LightningModule and LightningDataModule are two separate components with distinct responsibilities. Occasionally, conflicts in hyperparameters may arise between the model and the data...

Enhance model training by enabling Exponential Moving Average (EMA) on the model parameters. To activate this feature, simply specify the `ema_prmtrs_decay` arg in the configuration file. If not set, its...

Add backup for metrics.csv

1

We currently use `CSVLogger` from `torch.lightning` to write metrics to a CSV file in the log directory. However, when using `load_model` (i.e., loading from a checkpoint), or more in general...

This PR introduces a custom version of MDCATH while preserving the original functionality to ensure backward compatibility. New Arguments: - `source_file `and `file_basename`: these args enable users to create and...