【error】RuntimeError: size mismatch与全连接fc层
今天跑代码的时候遇到了这个错误:RuntimeError: size mismatch调试之后发现是如下有问题:源代码在这里:这是class里init中相应的部分def forward的相应的问题问题就在于y1 = self.fc_1(y)这里,fc_1也是同fc一样的全连接层,而全连接层输入的尺寸只有两维!从图上可以看到,y输出的尺寸是[2,576,1,1]在y1 = se...
原创
2019-03-07 16:35:20 ·
21725 阅读 ·
4 评论