提交 e1816227 编写于 作者: xuyang2233's avatar xuyang2233

modified rb head and infer_benchmark

上级 c53a6e51
...@@ -465,9 +465,6 @@ class PositionAttentionDecoder(BaseDecoder): ...@@ -465,9 +465,6 @@ class PositionAttentionDecoder(BaseDecoder):
_, c_feat, _, _ = feat.shape _, c_feat, _, _ = feat.shape
assert c_feat == self.dim_input 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) position_out_enc = self.position_aware_module(out_enc)
query = self.embedding(position_index) query = self.embedding(position_index)
......
...@@ -49,4 +49,6 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.t ...@@ -49,4 +49,6 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.t
--save_log_path:./test/output/ --save_log_path:./test/output/
--benchmark:True --benchmark:True
null:null null:null
===========================infer_benchmark_params==========================
random_infer_input:[{float32,[3,48,160]}]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册