Tensorflow-MKL 在 3D 数据上给出错误答案

【问题标题】：Tensorflow-MKL giving Errors on 3D dataTensorflow-MKL 在 3D 数据上给出错误
【发布时间】：2018-07-05 01:07:09
【问题描述】：

我正在训练一个模型来执行体积分割（3D 数据）。由于输入数据的大小，我正在 CPU 上进行训练。我正在使用带有 tensorflow-mkl 和 keras 的 anaconda 环境。当我训练模型时，我得到一个error：

"tensorflow.python.framework.errors_impl.InvalidArgumentError: Value for attr 'data_format' of "NDHWC" is not in the list of allowed values: "NHWC", "NCHW"

但是，在他们的 github 上，它说它适用于体积分割 (https://github.com/intel/mkl-dnn)。我该如何解决这个问题，以便我可以使用 mkl 训练我的 3D-Unet？

【问题讨论】：

标签： tensorflow keras deep-learning intel-mkl intel-dnn

【解决方案1】：

尝试将 tensorflow 更新到 1.8.0 版本。希望这能解决您的问题。

【讨论】：