赞
踩
今天在运行卷积神经网络时遇到如下错误:
ValueError: Dimensions must be equal, but are 32 and 3 for 'Conv2D_1' (op: 'Conv2D') with input shapes: [?,32,32,32], [3,3,3,32].
经过设置断点调试得知在进行第二层卷积的时候遇到了问题:
第二层卷积的输入值维度:x=[?,32,32,32]
第二层卷积的权重维度:W=[3,3,3,32]
在进行卷积操作时,输入值x的第四个维度值为通道数,权重的第三个值和x的第四个值应该是对应的,都是代表通道数。
把W改成[3,3,32,32]问题解决。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。