【error】RuntimeError: size mismatch与全连接fc层

今天跑代码的时候遇到了这个错误: RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 192] ,at /opt/conda/conda-bld/pytorch_1524 调试之后发现是如下有问题: 源代码在这里: 这是class里init中相应的部分 def forward的相应的问题 问题就在于y1 = self.fc_1(y)这里,fc
相关文章
相关标签/搜索