Thank you for your quick replying!
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 224, 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 201, 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)

Can’t it have something to do with the forward functions?