gluon-cv: Error Training DeepLab on mxnet-mkl
@zhreshold I’ve encountered an error trying to train deeplab on cpu using mxnet-mkl.
Steps to reproduce:
- Installing mxnet-mkl:
pip3 install mxnet-mkl - Using gluoncv’s train.py and running the following command :
python3 gluoncv_test.py --dataset pascal_aug --model-zoo deeplab_resnet101_coco --aux --lr 0.001 --checkname res101 --no-cuda - This will cause the following error:
mxnet.base.MXNetError: [15:37:48] src/ndarray/ndarray.cc:757: Check failed: !IsMKLDNNData(): We can't generate TBlob for MKLDNN data. Please use Reorder2Default() to generate a new NDArray first
About this issue
- Original URL
- State: closed
- Created 4 years ago
- Comments: 16 (6 by maintainers)
@wuxun-zhang @xinyu-intel @zhreshold Thanks for the fix, the problem is solved.
I think mxnet-mkl is no longer being updated for now. Please try to use mxnet directly since mkl-dnn is enabled by default. Try nightly build
pip install --pre mxnet -f https://dist.mxnet.io/python/cpu