Solved by Remorax in post #3 
              
              
                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/…