From 6ba3765564968ba55740e4b95026c6b4c44fdfcd Mon Sep 17 00:00:00 2001 From: Daniel Campos Date: Mon, 25 Oct 2021 09:35:36 -0500 Subject: [PATCH] changing qa to ner --- examples/pytorch/token-classification/run_ner.py | 2 +- examples/pytorch/token-classification/sparseml_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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