本文为总结自己在调试pytorch程序中遇到的问题和解决方法
- Linear()层维度不匹配
错误提示:
size mismatch, m1: [20 x 12800], m2: [15488x 784] at…
解决方法:
nn.Linear(15488, 784)时报错,因为输入的矩阵维度为12800,因此将语句改为nn.Linear(12800, 784)即可。m1对应的输入x的维度,m2为权重w的维度。
未完待续。。。
ps:希望这篇文章越短越好。
本文为总结自己在调试pytorch程序中遇到的问题和解决方法
size mismatch, m1: [20 x 12800], m2: [15488x 784] at…
解决方法:
nn.Linear(15488, 784)时报错,因为输入的矩阵维度为12800,因此将语句改为nn.Linear(12800, 784)即可。m1对应的输入x的维度,m2为权重w的维度。
未完待续。。。
ps:希望这篇文章越短越好。