From fcab661af9cb4461c996fb5476667ca90b70f3f0 Mon Sep 17 00:00:00 2001 From: Yu Chengzhang Date: Mon, 30 Jun 2025 19:51:07 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E4=BA=86=E9=85=8D=E7=BD=AE?= =?UTF-8?q?=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model/LMConfig.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/model/LMConfig.py b/model/LMConfig.py index f5e0012..505376d 100644 --- a/model/LMConfig.py +++ b/model/LMConfig.py @@ -41,6 +41,12 @@ class LMConfig(PretrainedConfig): knowledge_num: int = 64*64, knowledge_length: int = 8, knowledge_dim: int = 128, + #################################################### + # Triple extraction related configurations + #################################################### + max_subject_len: int = 8, + max_predicate_len: int = 4, + max_object_len: int = 8, **kwargs, ): self.dim = dim @@ -76,4 +82,10 @@ class LMConfig(PretrainedConfig): self.knowledge_num = knowledge_num self.knowledge_length = knowledge_length 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)