✏️ Fix typo (#5734)

This commit is contained in:
Cola 2020-07-27 23:55:15 +09:00 committed by GitHub
parent d1d15d6f2d
commit 5779e5434d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -575,4 +575,4 @@ class TFTrainer:
if not isinstance(self.model, TFPreTrainedModel):
raise ValueError("Trainer.model appears to not be a PreTrainedModel")
self.model.save_pretrained(self.args.output_dir)
self.model.save_pretrained(output_dir)