From 61abd50b98de4cd8f6024d77154a0326d16f1db5 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 9 Dec 2020 09:13:41 -0500 Subject: [PATCH] Remove use of deprected method in Trainer HP search (#8996) --- 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 90ded12f78..8fbe1729fa 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -543,7 +543,7 @@ class Trainer: self.args.output_dir = checkpoint_dir output_dir = os.path.join(self.args.output_dir, f"{PREFIX_CHECKPOINT_DIR}-{self.state.global_step}") self.save_model(output_dir) - if self.is_world_master(): + if self.is_world_process_zero(): self.state.save_to_json(os.path.join(output_dir, "trainer_state.json")) torch.save(self.optimizer.state_dict(), os.path.join(output_dir, "optimizer.pt")) torch.save(self.lr_scheduler.state_dict(), os.path.join(output_dir, "scheduler.pt"))