pytorch错误提示和应对方法总结（不定时更新）

本文为总结自己在调试pytorch程序中遇到的问题和解决方法

1. 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：希望这篇文章越短越好。