- 博客(2)
- 收藏
- 关注
原创 [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
2020-06-03 17:12:09 870
原创 [pytorch] TypeError cannot assign torch.FloatTensor as parameter weight
【pytorch】TypeError: cannot assign ‘torch.FloatTensor’ as parameter ‘weight’ (torch.nn.Parameter or None expected)在尝试赋值线性层权重时候出现的错误错误定位 def __init__(self, input_dim, atten_dim, attribute, classifier): super(MODEL,self).__init__() nc
2020-05-29 00:02:23 21929 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人