利用卷积搭建网络时 常常会报错ValueError: Depth of input (1) is not a multiple of input depth of filter (3) for '{{node inference/subnet1/sam/Conv2D}} = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], explicit_paddings=[], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true](inference/subnet1/Relu, w0/read)' with input shapes: [10,5,5,1], [3,3,3,1]. 我自己手写了部分卷积模块,然后报错 找了好久只发现一篇文章说的https://www.zhihu.com/question/263391986,不太懂 但我觉得是我的特征图在经过卷积后 由于维度不一致导致不能相乘,再努力改改吧 加油 小菜鸡
在tensorflow上搭建网络遇到的坑
最新推荐文章于 2024-07-20 16:37:13 发布