Default MaxPoolingOp only supports NHWC on device type CPU

tensorflow2.3
keras 2.4
cpu版本
出现Default MaxPoolingOp only supports NHWC on device type CPU这个错误,就是代码里面用了GPU版本的代码,所以报错。

我的解决办法:
data_format:可选"NHWC",“NCHW”,默认为"NHWC"。指定输入和输出数据的格式,默认值表示data将被存储为[batch, height, width, channels]的顺序。"NCHW"表示data将被存储为[batch,channels, height, width] 的顺序。

我使用的是data_format=‘channels_first’,不是默认值,所以是GPU版本的属性,所以报错,将其设置为默认即可。

你可能感兴趣的:(tensorflow)