Python TypeError - 'Class' object is not callable (Google Collab Example Inside)
Your UCC_Classifier model should be a pl.LightningModule, not a pl.LightningDataModule.
Python TypeError - 'Class' object is not callable (Google Collab Example Inside)
Your UCC_Classifier model should be a pl.LightningModule, not a pl.LightningDataModule.