SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix type annotation for debug arg (#24033)

Browse Source 
* fix type annotation for debug arg

* fix TypeErorr
This commit is contained in:
Bearnardd
2023-06-21 12:42:21 +02:00
committed by GitHub
parent 16c7b16a0a
commit 4c6e429589
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/training_args.py
View File

@@ -873,7 +873,7 @@ class TrainingArguments:
) )
}, },
) )
debug: str = field( debug: Union[str, List[DebugOption]] = field(
default="", default="",
metadata={ metadata={
"help": ( "help": (
@@ -1563,8 +1563,11 @@ class TrainingArguments:
) )
self.debug += " tpu_metrics_debug" self.debug += " tpu_metrics_debug"
self.tpu_metrics_debug = False self.tpu_metrics_debug = False
if isinstance(self.debug, str): if isinstance(self.debug, str):
self.debug = [DebugOption(s) for s in self.debug.split()] self.debug = [DebugOption(s) for s in self.debug.split()]
elif self.debug is None:
self.debug = []
self.deepspeed_plugin = None self.deepspeed_plugin = None
if self.deepspeed: if self.deepspeed: