check failed: (in_data[softmax_activation::kData].ndim()) == (2)
Hi ,@piiswrong [15:44:47] /home/git/mxnet/mxnet/dmlc-core/include/dmlc/logging.h:235: [15:44:47] src/operator/./cudnn_softmax_activation-inl.h:44: Check failed: (in_data[softmax_activation::kData].ndim()) == (2) Input need to have 2 dimensions when mode=instance.
[15:44:47] /home/git/mxnet/mxnet/dmlc-core/include/dmlc/logging.h:235: [15:44:47] src/engine/./threaded_engine.h:306: [15:44:47] src/operator/./cudnn_softmax_activation-inl.h:44: Check failed: (in_data[softmax_activation::kData].ndim()) == (2) Input need to have 2 dimensions when mode=instance.
cuda7.5+cudnn5.0
I get this result when I run "python ./train.py".
How to solve this problem, thank you
@xionglei181818 Hi, I met this problem too,Did you solve it? Could you tell me how to solver this problem? Thank you!
@xionglei181818 Hi i solved this problem. Because of the mxnet's version, you can change the the 'type' to 'mode ' in mx.symbol.SoftmaxActivation(....);
@lwkobe I met same problem, follow your method change 'type' to 'mode' doesn't solve it. My system is ubutu16.04 cuda8.0 cudnn5 mxnet0.9.3. this problem drive me crazy, can you help me out, thanks very much!