Tensorflow-MKL 给出 3D 数据错误
Tensorflow-MKL giving Errors on 3D data
我正在训练一个模型来执行体积分割（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到1.8.0版本。
希望这能解决您的问题。
我正在训练一个模型来执行体积分割（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?
尝试更新tensorflow到1.8.0版本。 希望这能解决您的问题。