search

Shivanandroy / simpleT5

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train your T5 models.
MIT License
386 stars 62 forks source link

CUDA error or IndexError (on CPU) #14

Open djstrong opened 3 years ago

djstrong commented 3 years ago

To reproduce run: https://colab.research.google.com/drive/1SgLlLBI16ZQkjMAjJhzfNhAO0D6zgUwh?usp=sharing Probably, it is a bug in transformers.