Keep the new hyperparameters when fine-tuning a model, fix: #1942

hankcs · Jan 14, 2025 · 80506cf · 80506cf
1 parent a639dbc
commit 80506cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hanlp/common/torch_component.py b/hanlp/common/torch_component.py
@@ -248,9 +248,9 @@ def fit(self,
             _dummy_placeholder = self._create_dummy_placeholder_on(first_device)
         if finetune:
             if isinstance(finetune, str):
-                self.load(finetune, devices=devices)
+                self.load(finetune, devices=devices, **self.config)
             else:
-                self.load(save_dir, devices=devices)
+                self.load(save_dir, devices=devices, **self.config)
             self.config.finetune = finetune
             self.vocabs.unlock()  # For extending vocabs
             logger.info(