diff --git a/classify_infer.py b/classify_infer.py index 6c88b5af4fd40ca50598fe175ac2447027baa56a..3aedce47581f47c1038ab7be2d906cb607e066e6 100644 --- a/classify_infer.py +++ b/classify_infer.py @@ -47,7 +47,7 @@ from finetune_args import parser # yapf: disable parser = argparse.ArgumentParser(__doc__) model_g = ArgumentGroup(parser, "model", "options to init, resume and save model.") -model_g.add_arg("ernie_config_path", str, None, "Path to the json file for bert model config.") +model_g.add_arg("ernie_config_path", str, None, "Path to the json file for ernie model config.") model_g.add_arg("init_checkpoint", str, None, "Init checkpoint to resume training from.") model_g.add_arg("save_inference_model_path", str, "inference_model", "If set, save the inference model to this path.") model_g.add_arg("use_fp16", bool, False, "Whether to resume parameters from fp16 checkpoint.") diff --git a/predict_classifier.py b/predict_classifier.py index df641dcdaa10d69e4492b1ed444366c628aac8de..67c219a350877c8a925b69f52dd79f92e0bf77b7 100644 --- a/predict_classifier.py +++ b/predict_classifier.py @@ -41,7 +41,7 @@ from finetune_args import parser # yapf: disable parser = argparse.ArgumentParser(__doc__) model_g = ArgumentGroup(parser, "model", "options to init, resume and save model.") -model_g.add_arg("ernie_config_path", str, None, "Path to the json file for bert model config.") +model_g.add_arg("ernie_config_path", str, None, "Path to the json file for ernie model config.") model_g.add_arg("init_checkpoint", str, None, "Init checkpoint to resume training from.") model_g.add_arg("save_inference_model_path", str, "inference_model", "If set, save the inference model to this path.") model_g.add_arg("use_fp16", bool, False, "Whether to resume parameters from fp16 checkpoint.")