We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7c829b6 commit 161241eCopy full SHA for 161241e
src/lightning/pytorch/plugins/precision/deepspeed.py
@@ -141,9 +141,9 @@ def optimizer_step( # type: ignore[override]
141
@override
142
def clip_gradients(
143
self,
144
- module: Optional[Module],
145
optimizer: Optimizer,
146
clip_val: Union[int, float] = 0.0,
147
gradient_clip_algorithm: GradClipAlgorithmType = GradClipAlgorithmType.NORM,
+ module: Optional[Module] = None,
148
) -> None:
149
"""DeepSpeed handles gradient clipping internally."""
0 commit comments