From 6af5a54c28c84e248dac2df2ad7b40968fded3da Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Thu, 23 Apr 2020 11:02:05 -0400 Subject: [PATCH] [Trainer] reuse constant --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index b6009a8106..891e70f0eb 100644 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -358,7 +358,7 @@ class Trainer: else: assert model is self.model # Save model checkpoint - output_dir = os.path.join(self.args.output_dir, f"checkpoint-{global_step}") + output_dir = os.path.join(self.args.output_dir, f"{PREFIX_CHECKPOINT_DIR}-{global_step}") self.save_model(output_dir) self._rotate_checkpoints() torch.save(optimizer.state_dict(), os.path.join(output_dir, "optimizer.pt"))