Fix tpu_metrics_debug (#24452)

fix for tpu metrics debugs string
2023-06-26 02:59:07 -07:00
parent 3b84d86b57
commit be2d9f2e47
1 changed files with 4 additions and 1 deletions
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -1561,6 +1561,9 @@ class TrainingArguments:
                " `--debug tpu_metrics_debug` instead",
                FutureWarning,
            )
+            if self.debug is None:
+                self.debug = " tpu_metrics_debug"
+            else:
                self.debug += " tpu_metrics_debug"
            self.tpu_metrics_debug = False