Fix tpu_metrics_debug (#24452)

fix for tpu metrics debugs string

Fix tpu_metrics_debug (#24452)
fix for tpu metrics debugs string
be2d9f2e · Meghan Cowan · GitHub · 3b84d86b · be2d9f2e
Unverified Commit be2d9f2e authored Jun 26, 2023 by Meghan Cowan Committed by GitHub Jun 26, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/transformers/training_args.py src/transformers/training_args.py +4 -1

No files found.
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -1561,7 +1561,10 @@ class TrainingArguments:
                " `--debug tpu_metrics_debug` instead",
                FutureWarning,
            )
-            self.debug += " tpu_metrics_debug"
+            if self.debug is None:
+                self.debug = " tpu_metrics_debug"
+            else:
+                self.debug += " tpu_metrics_debug"
            self.tpu_metrics_debug = False

        if isinstance(self.debug, str):