@Nikronic I don’t think the tensor in question is already on GPU, since it is of type torch.FloatTensor and not torch.cuda.FloatTensor.
It would be helpful to print the full stacktrace, so we can at least know what line throws the error!