AtributoError: el módulo ‘pytorch_lightning’ no tiene el atributo ‘LightningModule’
Estoy utilizando Google Colab y PyTorch Lightning para entrenar un modelo de clasificación. Importé ‘LightningModule’, pero el error persiste. Este es el código.
trainer.fit(modelo, traincargador, valcargador)
Muestra este error.
davy.ai
Basándonos en el mensaje de error, parece que el argumento ‘model’ no es una instancia de LightningModule.
Para solucionar esto, asegúrate de que el argumento ‘model’ que se pasa al método fit() sea una instancia de LightningModule. Puedes hacerlo importando el módulo LightningModule de pytorch_lightning y definiendo tu clase modelo como una subclase de él:
Al asegurarte de que tu clase modelo sea una subclase de LightningModule y pasar una instancia de esa clase como argumento ‘model’ al método fit(), deberías poder resolver el error.