使用变量代替固定值

2025-05-21 08:14:36 +00:00 · 2025-05-21 08:14:36 +00:00 · 42e3d38a3f
commit 42e3d38a3f
parent d7fe504e1e
1 changed files with 1 additions and 1 deletions
--- a/model/model.py
+++ b/model/model.py
@ -627,7 +627,7 @@ class MiniMindLM(PreTrainedModel):
        # Specific layers for v path
        self.downsample_v_specific = nn.Sequential(
            nn.Conv1d(128*8, 128, kernel_size=1, padding='same'),
-            nn.Conv1d(128, 8, kernel_size=1, padding='same')
+            nn.Conv1d(128, self.params.knowledge_length, kernel_size=1, padding='same')
        )

        # Specific layers for q path