标签:dim py mat2 nn mat1 self torch forward
PyTorch错误 torch.addmm RuntimeError: mat1 dim 1 must match mat2 dim 0
flyfish
详细提示
File "python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/media/ubuntu/data/pytorch1.7/soothsayer/model/net_regression.py", line 50, in forward
x = self.predict(x)
File "python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "python3.6/site-packages/torch/nn/modules/linear.py", line 93, in forward
return F.linear(input, self.weight, self.bias)
File "python3.6/site-packages/torch/nn/functional.py", line 1690, in linear
ret = torch.addmm(bias, input, weight.t())
RuntimeError: mat1 dim 1 must match mat2 dim 0
代码
self.fc1 = nn.Linear(128, 120)
self.fc2 = nn.Linear(120, 84)
假设forward中要依次执行fc1和fc2
fc1的out_features 等于fc2的in_features就不会有错误,这里都是120不会有错误
反之
self.fc1 = nn.Linear(128, 121)
self.fc2 = nn.Linear(120, 84)
则出现让面的错误提示
标签:dim,py,mat2,nn,mat1,self,torch,forward 来源: https://blog.csdn.net/flyfish1986/article/details/115263892
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。