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 94a5ab7 commit 876703aCopy full SHA for 876703a
pytorch_lightning/utilities/distributed.py
@@ -244,7 +244,7 @@ def all_gather_ddp_if_available(
244
245
246
def register_ddp_comm_hook(
247
- model: Type[DistributedDataParallel],
+ model: DistributedDataParallel,
248
ddp_comm_state: Optional[object] = None,
249
ddp_comm_hook: Optional[Callable] = None,
250
ddp_comm_wrapper: Optional[Callable] = None,
0 commit comments