diff options
| author |   HsiangNianian <i@jyunko.cn> | 2025-12-30 20:39:34 +0800 |
|---|---|---|
| committer | HsiangNianian <i@jyunko.cn> | 2025-12-30 20:39:34 +0800 |
| commit | 298035052b3e3d083b57f5dbac0e86de4f94efba (patch) | |
| tree | 944f38d734f752a5a0f71033ebece38fc5c35839 /src/base_model_trpgner/training/__init__.py | |
| parent | 92a647ffbb3452a0ed49601177f290e20a88413e (diff) | |
| download | base-model-298035052b3e3d083b57f5dbac0e86de4f94efba.tar.gz base-model-298035052b3e3d083b57f5dbac0e86de4f94efba.zip | |
refactor: Update model download functionality and improve inference module to support automatic model retrieval from GitHub releases
Diffstat (limited to 'src/base_model_trpgner/training/__init__.py')
| -rw-r--r-- | src/base_model_trpgner/training/__init__.py | 4 |
1 files changed, 2 insertions, 2 deletions
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: |