InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device ty

InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device type CPU
[[Node: ssd_300_vgg/pool1/MaxPool = MaxPoolT=DT_FLOAT, data_format="NCHW", ksize=[1, 1, 2, 2], padding="SAME", strides=[1, 1, 2, 2], _device="/job:localhost/replica:0/task:0/device:CPU:0"]] 
解决方法:you can modify the code 'NCHW' as 'NHWC'
参考链接:https://github.com/balancap/SSD-Tensorflow/issues/226

你可能感兴趣的:(InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device ty)