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

Deprecate model_path in Trainer.train (#9854)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-01-28 08:32:46 -05:00
committed by GitHub
parent 2ee9f9b69e
commit b4e559cfa1
14 changed files with 96 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/seq2seq/run_seq2seq.py
View File

@@ -491,12 +491,12 @@ def main():
# Training
if training_args.do_train:
if last_checkpoint is not None:
model_path = last_checkpoint
checkpoint = last_checkpoint
elif os.path.isdir(model_args.model_name_or_path):
model_path = model_args.model_name_or_path
checkpoint = model_args.model_name_or_path
else:
model_path = None
train_result = trainer.train(model_path=model_path)
checkpoint = None
train_result = trainer.train(resume_from_checkpoint=checkpoint)
trainer.save_model() # Saves the tokenizer too for easy upload
output_train_file = os.path.join(training_args.output_dir, "train_results.txt")