"vscode:/vscode.git/clone" did not exist on "5ea2595ecde9a89079d4d289bc65dc5f8dc02ef6"
Unverified Commit 250032e9 authored by Costa Huang's avatar Costa Huang Committed by GitHub
Browse files

Minor type annotation fix (#27276)

* Minor type annotation fix

* Trigger Build
parent a53a0c51
......@@ -1144,7 +1144,7 @@ class TrainingArguments:
"help": "If True, use gradient checkpointing to save memory at the expense of slower backward pass."
},
)
gradient_checkpointing_kwargs: dict = field(
gradient_checkpointing_kwargs: Optional[dict] = field(
default=None,
metadata={
"help": "Gradient checkpointing key word arguments such as `use_reentrant`. Will be passed to `torch.utils.checkpoint.checkpoint` through `model.gradient_checkpointing_enable`."
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment