Можно ли использовать формат NCHW на процессоре?

Question

Я пытаюсь использовать NCHW, т. Е. Формат данных первого канала в моем процессоре. Это слой maxpool как часть Resnet18.

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

Можете ли вы попробовать простую модель для устранения проблемы?Это работает на моей системе с процессором.

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)