Skip to content

Commit 181a355

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent eea0a94 commit 181a355

File tree

1 file changed

+3
-1
lines changed
  • src/lightning/pytorch/plugins/precision

1 file changed

+3
-1
lines changed

src/lightning/pytorch/plugins/precision/fsdp.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,9 @@ def convert_module(self, module: Module) -> Module:
8282
return module
8383

8484
@override
85-
def clip_grad_by_norm(self, optimizer: Optimizer, clip_val: Union[int, float], module: Optional[Module] = None) -> None:
85+
def clip_grad_by_norm(
86+
self, optimizer: Optimizer, clip_val: Union[int, float], module: Optional[Module] = None
87+
) -> None:
8688
# see https://pytorch.org/docs/stable/fsdp.html#torch.distributed.fsdp.FullyShardedDataParallel.clip_grad_norm_
8789
if module is None:
8890
return

0 commit comments

Comments
 (0)