iomgaa/Minimind
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新了配置文件

Browse Source
This commit is contained in:
Yu Chengzhang 2025-06-30 19:51:07 +08:00
parent 31eff0e3de
commit fcab661af9
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
model/LMConfig.py
View File

@ -41,6 +41,12 @@ class LMConfig(PretrainedConfig):
knowledge_num: int = 64*64, knowledge_num: int = 64*64,
knowledge_length: int = 8, knowledge_length: int = 8,
knowledge_dim: int = 128, knowledge_dim: int = 128,
####################################################
# Triple extraction related configurations
####################################################
max_subject_len: int = 8,
max_predicate_len: int = 4,
max_object_len: int = 8,
**kwargs, **kwargs,
): ):
self.dim = dim self.dim = dim
@ -76,4 +82,10 @@ class LMConfig(PretrainedConfig):
self.knowledge_num = knowledge_num self.knowledge_num = knowledge_num
self.knowledge_length = knowledge_length self.knowledge_length = knowledge_length
self.knowledge_dim = knowledge_dim self.knowledge_dim = knowledge_dim
####################################################
# Triple extraction related configurations
####################################################
self.max_subject_len = max_subject_len
self.max_predicate_len = max_predicate_len
self.max_object_len = max_object_len
super().__init__(**kwargs) super().__init__(**kwargs)