RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument
By Thankful Tarantula at Jun 06 2021