From a753cafdc0aed41c8385298ace3eca4c213c556d Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 26 Apr 2021 08:37:32 -0700 Subject: [PATCH] [docs] fix invalid class name (#11438) * fix invalid class name * proper ref * proper ref --- docs/source/migration.md | 4 ++-- src/transformers/integrations.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/source/migration.md b/docs/source/migration.md index e44af20c0f..7b97867e33 100644 --- a/docs/source/migration.md +++ b/docs/source/migration.md @@ -169,8 +169,8 @@ Regarding the `TFTrainer` class: - The `TFTrainer` method `_setup_wandb` is deprecated in favor of `setup_wandb`. - The `TFTrainer` method `_run_model` is deprecated in favor of `run_model`. -Regarding the `TrainerArgument` class: -- The `TrainerArgument` argument `evaluate_during_training` is deprecated in favor of `evaluation_strategy`. +Regarding the `TrainingArguments` class: +- The `TrainingArguments` argument `evaluate_during_training` is deprecated in favor of `evaluation_strategy`. Regarding the Transfo-XL model: - The Transfo-XL configuration attribute `tie_weight` becomes `tie_words_embeddings`. diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py index 7e4ab0f5c7..9ab198cf14 100644 --- a/src/transformers/integrations.py +++ b/src/transformers/integrations.py @@ -856,8 +856,8 @@ class MLflowCallback(TrainerCallback): Whether to use MLflow .log_artifact() facility to log artifacts. This only makes sense if logging to a remote server, e.g. s3 or GCS. If set to `True` or `1`, will copy - whatever is in TrainerArgument's output_dir to the local or remote artifact storage. Using it without a - remote storage will just copy the files to your artifact location. + whatever is in :class:`~transformers.TrainingArguments`'s ``output_dir`` to the local or remote + artifact storage. Using it without a remote storage will just copy the files to your artifact location. """ log_artifacts = os.getenv("HF_MLFLOW_LOG_ARTIFACTS", "FALSE").upper() if log_artifacts in {"TRUE", "1"}: