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

2026-02-14 15:29:25 +08:00 · 2026-02-14 15:29:25 +08:00 · b68f75b09d
parent d2d65c7efa
commit b68f75b09d
1 changed files with 1 additions and 1 deletions
--- a/src/suinput/dataset.py
+++ b/src/suinput/dataset.py
@ -440,7 +440,7 @@ class PinyinInputDataset(IterableDataset):
                "char_id": torch.tensor([char_info["id"]]),
                "char": char,
                "freq": char_info["freq"],
-                "pg": torch.tensor([self.pg_groups[char_info.pinyin[0]]]),
+                "pg": torch.tensor([self.pg_groups[char_info["pinyin"][0]]]),
            }

            # 根据调整因子重复样本