diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index c57fcafdfcc6..2c3d9e2cd866 100644 --- a/src/transformers/training_args.py +++ b/src/transformers/training_args.py @@ -885,6 +885,12 @@ class TrainingArguments: ) }, ) + logging_dir: Optional[str] = field( + default=None, + metadata={ + "help": "Deprecated and will be removed in v5.2. Set env var `TENSORBOARD_LOGGING_DIR` instead. TensorBoard log directory." + }, + ) logging_strategy: Union[IntervalStrategy, str] = field( default="steps", metadata={"help": "The logging strategy to use."}, @@ -1695,6 +1701,11 @@ def __post_init__(self): if isinstance(self.include_num_input_tokens_seen, bool): self.include_num_input_tokens_seen = "all" if self.include_num_input_tokens_seen else "no" + if self.logging_dir is not None: + logger.warning( + "`logging_dir` is deprecated and will be removed in v5.2. Please set `TENSORBOARD_LOGGING_DIR` instead." + ) + def __str__(self): self_as_dict = asdict(self)