ThilinaRajapakse/simpletransformers

RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

celsowm opened this issue · 0 comments

celsowm commented

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):

  • OS: Windows 10
  • Pytorch: 2.1.0+cu121
  • Python: 3.10.9 (tags/v3.10.9:1dd9be6, Dec 6 2022, 20:01:21) [MSC v.1934 64 bit (AMD64)]