Device index must not be negative

[jaimeraynaud]

Hello, I am asking for help to solve the error that is giving me when executing the code. I attach a screenshot of it:

For what I see in https://stackoverflow.com/questions/60713781/pytorch-device-and-todevice-method they recommend not using get_device(), but I do not know how to solve it. Thanks in advance.