✏️ Fix typo (#5734)

2020-07-27 10:55:15 -04:00
parent d1d15d6f2d
commit 5779e5434d
1 changed files with 1 additions and 1 deletions
@@ -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)