From 745ea78dcc907fe47ea752805529719132c53d44 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Thu, 4 Mar 2021 09:44:02 -0500 Subject: [PATCH] Revert "Not always consider a local model a checkpoint in run_glue" This reverts commit f3660613bc14188e04e8eb4e27ae97f57b6b92d6. --- examples/text-classification/run_glue.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/examples/text-classification/run_glue.py b/examples/text-classification/run_glue.py index 28a33091f74..cc8ea89f052 100755 --- a/examples/text-classification/run_glue.py +++ b/examples/text-classification/run_glue.py @@ -406,15 +406,12 @@ def main(): # Training if training_args.do_train: - checkpoint = None if last_checkpoint is not None: checkpoint = last_checkpoint elif os.path.isdir(model_args.model_name_or_path): - # Check the config from that potential checkpoint has the right number of labels before using it as a - # checkpoint. - if AutoConfig.from_pretrained(model_args.model_name_or_path).num_labels == num_labels: - checkpoint = model_args.model_name_or_path - + checkpoint = model_args.model_name_or_path + else: + checkpoint = None train_result = trainer.train(resume_from_checkpoint=checkpoint) metrics = train_result.metrics