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
最新推荐文章于 2023-05-30 23:31:30 发布