Closed FrancescoSaverioZuppichini closed 1 year ago
Following #6, I've made the LightningModule taking the model and the loss function, making it easier to swap them later on (assuming they follow some interface I still need to code).
LightningModule
Description
Following #6, I've made the
LightningModuletaking the model and the loss function, making it easier to swap them later on (assuming they follow some interface I still need to code).