NCHW和CHW有啥区别?
收藏
NCHW和CHW有啥区别?
ValueError: (InvalidArgument) The number of input's channels should be equal to filter's channels * groups for Op(Conv). But received: the input's channels is 224, the input's shape is [64, 224, 224, 3]; the filter's channels is 3, the filter's shape is [64, 3, 7, 7]; the groups is 1, the data_format is NCHW. The error may come from wrong data_format setting.
0
收藏
请登录后评论
N是batch size
送入model的一般都是NCHW形状的,即使是[1,C,H,W]
动态图里跑的是NCHW数据,你自己的图像数据是HWC,CHW的挺少见。。。
对对对,C跑后面去了。
当时我也给这个搞迷糊过,不过动态图能打印,忘记了就print一下
你不是问的“NCHW和CHW有啥区别”么。。。
其实就多了个维度呀,哈哈
C维度也挺调皮啊~
少个字母的区别