From d018afced056fe384de066ef2f0ed0f9e1458561 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 4 Jan 2021 07:14:32 -0800 Subject: [PATCH] [trainer] parametrize default output_dir (#9352) This PR: * fixes trainer to have the logger agree with the actual default `output_dir`, but setting it one place and passing it as an argument to both places @sgugger --- src/transformers/trainer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 6756e59165..d76ba6be7b 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -228,8 +228,9 @@ class Trainer: optimizers: Tuple[torch.optim.Optimizer, torch.optim.lr_scheduler.LambdaLR] = (None, None), ): if args is None: - logger.info("No `TrainingArguments` passed, using the current path as `output_dir`.") - args = TrainingArguments("tmp_trainer") + output_dir = "tmp_trainer" + logger.info(f"No `TrainingArguments` passed, using `output_dir={output_dir}`.") + args = TrainingArguments(output_dir=output_dir) self.args = args # Seed must be set before instantiating the model when using model set_seed(self.args.seed)