Я тренирую модель для выполнения объемной сегментации (трехмерные данные). Я тренируюсь на CPU из-за размера входных данных. Я использую среду анаконды с tenorflow-MKL и Keras. Когда я тренирую модель, я получаю 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). Как я могу решить эту проблему, чтобы я мог тренировать свой 3D-Unet с mkl
?