Sorry, I posted the thread out of frustration and didn’t really think about cleaning it up.
Does that look better now? All of the tensors that I can think of that are necessary have had the to(device) performed on them as is suggested in this article and to me anyway, thats pretty much my code looks.
https://towardsdatascience.com/pytorch-switching-to-the-gpu-a7c0b21e8a99 2
Any help is much appreciated!