mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-24 23:33:08 +08:00
modified rb head and infer_benchmark
This commit is contained in:
@@ -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]}]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user