From 298035052b3e3d083b57f5dbac0e86de4f94efba Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Tue, 30 Dec 2025 20:39:34 +0800 Subject: refactor: Update model download functionality and improve inference module to support automatic model retrieval from GitHub releases --- src/base_model_trpgner/training/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/base_model_trpgner/training') diff --git a/src/base_model_trpgner/training/__init__.py b/src/base_model_trpgner/training/__init__.py index ccf3c03..4f8e30d 100644 --- a/src/base_model_trpgner/training/__init__.py +++ b/src/base_model_trpgner/training/__init__.py @@ -36,11 +36,11 @@ def train_ner_model( resume_from_checkpoint: 恢复检查点路径 Examples: - >>> from basemodeltrpgner.training import train_ner_model + >>> from base_model_trpgner.training import train_ner_model >>> train_ner_model( ... conll_data="./data", ... output_dir="./my_model", - ... epochs=10 + ... num_train_epochs=10 ... ) """ try: -- cgit v1.2.3-70-g09d2