diff --git a/examples/pytorch/token-classification/run_ner.py b/examples/pytorch/token-classification/run_ner.py index 22a32e09025c..eac5267efe94 100755 --- a/examples/pytorch/token-classification/run_ner.py +++ b/examples/pytorch/token-classification/run_ner.py @@ -61,7 +61,7 @@ class ModelArguments: metadata={"help": "Path to pretrained model or model identifier from huggingface.co/models"} ) distill_teacher: Optional[str] = field( - default=None, metadata={"help": "Teacher model which needs to be a trained QA model"} + default=None, metadata={"help": "Teacher model which needs to be a trained NER model"} ) config_name: Optional[str] = field( default=None, metadata={"help": "Pretrained config name or path if not the same as model_name"} diff --git a/examples/pytorch/token-classification/sparseml_utils.py b/examples/pytorch/token-classification/sparseml_utils.py index 200c68d680a2..0fafb0213584 100644 --- a/examples/pytorch/token-classification/sparseml_utils.py +++ b/examples/pytorch/token-classification/sparseml_utils.py @@ -8,7 +8,7 @@ class TokenClassificationModuleExporter(ModuleExporter): """ - Module exporter class for Question Answering + Module exporter class for Token Classification """ @classmethod