仅作为记录,大佬请跳过。
今天博主在跑大佬博主的pytorch cnn分类mnist数据集的程序——大佬博主文章(第四个网络)
出现问题:
经过尝试,发现是在定义网络中的问题
将x = x.view(128,-1)改为x = x.view(-1,64*3*3)
即可。
最后,再次感谢强大的大佬博主
仅作为记录,大佬请跳过。
今天博主在跑大佬博主的pytorch cnn分类mnist数据集的程序——大佬博主文章(第四个网络)
出现问题:
经过尝试,发现是在定义网络中的问题
将x = x.view(128,-1)改为x = x.view(-1,64*3*3)
即可。
最后,再次感谢强大的大佬博主