Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’
原因:模型和数据未全都放入GPU
问题解决:RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’
原因:模型和数据未全都放入GPU
问题解决:RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’