songsenand
/
SUInput
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复 char_info.pinyin 访问方式，使用字典形式确保兼容性

This commit is contained in:
songsenand 2026-02-14 15:29:25 +08:00
parent d2d65c7efa
commit b68f75b09d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/suinput/dataset.py
View File

@ -440,7 +440,7 @@ class PinyinInputDataset(IterableDataset):
"char_id": torch.tensor([char_info["id"]]), "char_id": torch.tensor([char_info["id"]]),
"char": char, "char": char,
"freq": char_info["freq"], "freq": char_info["freq"],
"pg": torch.tensor([self.pg_groups[char_info.pinyin[0]]]), "pg": torch.tensor([self.pg_groups[char_info["pinyin"][0]]]),
} }
# 根据调整因子重复样本 # 根据调整因子重复样本