tensorflow2.0 - 如何将保存的模型从 NCHW 转换为 NHWC？

Question

我在 GPU 上训练了我的 NCHW 模型并保存了最佳状态。
我现在想对 CPU 进行推断，它显然只支持 NHWC(我得到了一个错误提及)。
我是否必须使用 NHWC 重新训练我的模型，或者有没有办法转换我的模型？

Answer 1

我处于同样的情况，在尝试运行时看到这样的错误 model.predict在只有 CPU 可用的实例上的 GPU 训练模型上:

tensorflow.python.framework.errors_impl.InvalidArgumentError: Default MaxPoolingOp only supports NHWC on device type CPU

我最终发现，在 Intel CPU 上，只要启用 MKL，就可以成功地将模型应用于 NCHW 格式的数据。使用 pip，可以安装启用 MKL 的 tensorflow:

pip install intel-tensorflow

您可以使用以下命令检查它是否已启用(在 tensorflow 2.3 中):

tf.python._pywrap_util_port.IsMklEnabled()