是否可以在 CPU 上使用 NCHW 格式？

Question

我正在尝试在我的 cpu 中使用 NCHW 即通道优先数据格式。它是作为 Resnet18 一部分的 maxpool 层。

MaxPooling2D(pool_size=[3, 3], strides=2, padding='same', data_format='channels_first')

我得到的错误是：

InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device type CPU
     [[Node: max_pooling2d_3/MaxPool = MaxPool[T=DT_FLOAT, data_format="NCHW", ksize=[1, 1, 3, 3], padding="SAME", strides=[1, 1, 2, 2], _device="/job:localhost/replica:0/task:0/device:CPU:0"](batch_normalization_51/cond/Merge)]]

有没有办法解决这个问题？我也试过data_format="NCHW"，但它给出了同样的错误。

Answer 1

您能否尝试使用一个简单的模型来调试问题？这适用于我的 CPU 系统。

model = Sequential()
model.add(MaxPooling2D(pool_size=[3, 3], strides=2, padding='same', 
                    data_format='channels_first', input_shape=(3,224,224)))
model.summary()
X = np.random.randn(1,3,224,224)
Y = model.predict(X)
print(Y.shape)
(1, 3, 112, 112)

Answer 2

pip install intel-tensorflow

解决了问题，但训练似乎比以前慢了。