deep3d icon indicating copy to clipboard operation
deep3d copied to clipboard

check failed: (in_data[softmax_activation::kData].ndim()) == (2)

Open lei-xiong opened this issue 9 years ago • 3 comments

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

lei-xiong avatar Dec 19 '16 07:12 lei-xiong

@xionglei181818 Hi, I met this problem too,Did you solve it? Could you tell me how to solver this problem? Thank you!

lwkobe avatar Dec 24 '16 14:12 lwkobe

@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 avatar Dec 24 '16 18:12 lwkobe

@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!

tpys avatar Apr 11 '17 08:04 tpys