Both Dwight and ptrblck thank you very much for your replies!
I’ve tried this suggestion (but with all suggestions above I get the same result anyways:
Traceback (most recent call last):
  File "<input>", line 343, in <module>
  File "C:\Users\nero-\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "<input>", line 222, in forward
  File "C:\Users\nero-\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "<input>", line 165, in forward
RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

I’ve added a link to a small sample of the input file, I do not know if you have time to do such an exercise, but the moment model.to(device) is deleted if runs on the CPU (with such a small sample relatively fast). Data