[pytorch] Expected torch.LongTensor but found type torch.cuda.LongTensor for argument#4
RuntimeError: Expected object of type torch.LongTensor but found type torch.cuda.LongTensor for argument #4 ‘other’错误定位predicted_nclass = torch.LongTensor(1800, 85)output_avg = (output[0][0] + output[0][1] + output[0][2]) / 3predicted_nclass[start:end