diff --git a/ppocr/modeling/heads/rec_robustscanner_head.py b/ppocr/modeling/heads/rec_robustscanner_head.py index b9f8962d5a16c3c5dad16bd2b20e6f6d559a5f1c..7956059ecfe01f27db364d3d748d6af24dad0aac 100644 --- a/ppocr/modeling/heads/rec_robustscanner_head.py +++ b/ppocr/modeling/heads/rec_robustscanner_head.py @@ -465,9 +465,6 @@ class PositionAttentionDecoder(BaseDecoder): _, c_feat, _, _ = feat.shape assert c_feat == self.dim_input - # the _get_position_index is not ok for export_model - # position_index = self._get_position_index(self.max_seq_len, n) - position_out_enc = self.position_aware_module(out_enc) query = self.embedding(position_index) diff --git a/test_tipc/configs/rec_r31_robustscanner/train_infer_python.txt b/test_tipc/configs/rec_r31_robustscanner/train_infer_python.txt index 2f58d8f3e407bd3bc4431c7f56a81a42cae44584..07498c9e81ada9652343b8d8fff0f102d4684380 100644 --- a/test_tipc/configs/rec_r31_robustscanner/train_infer_python.txt +++ b/test_tipc/configs/rec_r31_robustscanner/train_infer_python.txt @@ -49,4 +49,6 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.t --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,48,160]}]