PaddleCustomDevice icon indicating copy to clipboard operation
PaddleCustomDevice copied to clipboard

请问PaddleCustom-mlu 哪个版本支持rnn,lstm

Open wangzy0327 opened this issue 1 year ago • 2 comments

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)

wangzy0327 avatar Jan 02 '25 07:01 wangzy0327

目前这个算子有限制,只支持lstm, 并且num_layer=1。 对ppocr/modeling/necks/rnn.py做如下修改: image

PeiyuLau avatar Jan 03 '25 01:01 PeiyuLau

目前这个算子有限制,只支持lstm, 并且num_layer=1。 对ppocr/modeling/necks/rnn.py做如下修改: image

请问仅支持lstm是Paddle框架的限制么?还是mlu库接口的问题呢?pytorch-mlu是支持rnn和lstm的

wangzy0327 avatar Jan 03 '25 01:01 wangzy0327