modified rb head and infer_benchmark

This commit is contained in:
smilelite
2022-07-13 22:33:11 +08:00
parent c53a6e5137
commit e181622756
2 changed files with 2 additions and 3 deletions
@@ -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)
@@ -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]}]