SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Not always consider a local model a checkpoint in run_glue

Browse Source
This commit is contained in:
Sylvain Gugger
2021-03-04 09:44:02 -05:00
parent 948b730f97
commit f3660613bc
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/text-classification/run_glue.py
View File

@@ -406,12 +406,15 @@ def main():
# Training # Training
if training_args.do_train: if training_args.do_train:
checkpoint = None
if last_checkpoint is not None: if last_checkpoint is not None:
checkpoint = last_checkpoint checkpoint = last_checkpoint
elif os.path.isdir(model_args.model_name_or_path): elif os.path.isdir(model_args.model_name_or_path):
checkpoint = model_args.model_name_or_path # Check the config from that potential checkpoint has the right number of labels before using it as a
else: # checkpoint.
checkpoint = None if AutoConfig.from_pretrained(model_args.model_name_or_path).num_labels == num_labels:
checkpoint = model_args.model_name_or_path
train_result = trainer.train(resume_from_checkpoint=checkpoint) train_result = trainer.train(resume_from_checkpoint=checkpoint)
metrics = train_result.metrics metrics = train_result.metrics