ThilinaRajapakse / simpletransformers

Transformers for Information Retrieval, Text Classification, NER, QA, Language Modelling, Language Generation, T5, Multi-Modal, and Conversational AI
https://simpletransformers.ai/
Apache License 2.0
4.11k stars 727 forks source link

RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int' #1553

Open celsowm opened 1 year ago

celsowm commented 1 year ago

Describe the bug

Tried to execute this code: https://simpletransformers.ai/docs/t5-minimal-start/ And got this error: RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

To Reproduce

Copy, paste on Thonny and executed

Expected behavior

To worked fine :)

Screenshots

image

Desktop (please complete the following information):