关于#pytorch#的问题:pytorch,报错找不到问题源头

img

img

img


本人初学pytorch,跟着视频教程打的代码,不知道问题出在了哪里😭麻烦大家看一下下,感谢感谢

dataloader中需要设置参数drop_last=True,因为你样本的总数不是批次的倍数,有余数存在,所以最后一个批次会出现错误,设置丢弃掉最后一个次数不足的批次就可以了,或者你reshape平均分布一下

报错问题应该是卷积的问题,卷积的两个输入尺寸不一致导致的,input是6416 ,weight是102464的,没法相乘报错。