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.
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.