PaddleCustomDevice
PaddleCustomDevice copied to clipboard
请问PaddleCustom-mlu 哪个版本支持rnn,lstm
ValueError: (InvalidArgument) MLU only support LSTM mode now, current mode is RNN_TANH [Hint: Expected mode == "LSTM", but received mode:RNN_TANH != "LSTM":LSTM.] (at /paddle/backends/mlu/kernels/rnn_kernel.cc:127)
ValueError: (InvalidArgument) MLU only support 1 num_layers, current num_layers is 3 [Hint: Expected num_layers == 1, but received num_layers:3 != 1:1.] (at /paddle/backends/mlu/kernels/rnn_kernel.cc:133)
目前这个算子有限制,只支持lstm, 并且num_layer=1。
对ppocr/modeling/necks/rnn.py做如下修改:
目前这个算子有限制,只支持lstm, 并且num_layer=1。 对ppocr/modeling/necks/rnn.py做如下修改:
请问仅支持lstm是Paddle框架的限制么?还是mlu库接口的问题呢?pytorch-mlu是支持rnn和lstm的
