"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "f4a18852983b8737368c35c9cfd3e3762b48c782"
Unverified Commit 0d74b46c authored by xiaoting's avatar xiaoting Committed by GitHub
Browse files

Merge pull request #1980 from tink2123/fix_encode_dygraph

Fix srn attn encoder for dygraph
parents 895d44bc 3d8b42dc
...@@ -215,7 +215,7 @@ class AttnLabelEncode(BaseRecLabelEncode): ...@@ -215,7 +215,7 @@ class AttnLabelEncode(BaseRecLabelEncode):
return None return None
data['length'] = np.array(len(text)) data['length'] = np.array(len(text))
text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len
- len(text) - 1) - len(text) - 2)
data['label'] = np.array(text) data['label'] = np.array(text)
return data return data
...@@ -261,7 +261,7 @@ class SRNLabelEncode(BaseRecLabelEncode): ...@@ -261,7 +261,7 @@ class SRNLabelEncode(BaseRecLabelEncode):
if len(text) > self.max_text_len: if len(text) > self.max_text_len:
return None return None
data['length'] = np.array(len(text)) data['length'] = np.array(len(text))
text = text + [char_num] * (self.max_text_len - len(text)) text = text + [char_num - 1] * (self.max_text_len - len(text))
data['label'] = np.array(text) data['label'] = np.array(text)
return data return data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment