Hi,
Thanks for getting back to me!
Fortunately, I was able to resolve this issue on my own a few minutes ago. I simply shifted the training data-to-Tensor conversion bit to the training loop, outside the “forward” function whilst keeping everything else the same.
For others who may face the same/similar issues in the future, just a tip: try to shift the conversion of input data to the training loop and push all tensors created in the “forward” function to “device”, as usual.
Thanks again!