diff options
Diffstat (limited to 'src/base_model_trpgner/training')
| -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: |