AttributeError: 'LightningDataModule' object has no attribute '_has_setup_TrainerFn.FITTING'

Shivanandroy / simpleT5

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train your T5 models.

MIT License

382 stars 61 forks source link

AttributeError: 'LightningDataModule' object has no attribute '_has_setup_TrainerFn.FITTING' #61

Open lucbuijs opened 4 months ago

lucbuijs commented 4 months ago

I'm trying to train a SimpleT5 model, but I keep getting this error. I'm running the code in Ubuntu (Linux). These are the versions I have: numpy 1.25.1 pandas 2.0.3 sentencepiece 0.2.0 torch 2.2.2 transformers 4.16.2 pytorch-lightning 1.5.10

NAMANVIKRAM commented 3 months ago

Try to run on google colab with enabling T4 GPU(by changing the runtime)