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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值