Pytorch错误总结 持续更新
类型不匹配Torch.cuda.FloatTensor & Torch.FloatTensor
输入类型是torch.cuda.FloatTensor,而weight type类型是torch.FloatTensor,两者类型不一致所以报错
原因: model没有放入cuda
解决办法:
net = xxxmodel()
if torch.cuda.is_available():
net = net.cuda()
NotImplementedError
**原因:**大概是没能找到forward函数,(我forward函数好像缩进在init函数里了,所以它没找到,有点愚蠢)
解决办法: 把forward函数以正确的格式缩进,能让编译器识别就行了。