Searched defs:FullyShardedDataParallel (Results 1 – 3 of 3) sorted by relevance
53 def register_fsdp(self, fsdp: FullyShardedDataParallel) -> None:78 def register_fsdp(self, fsdp: FullyShardedDataParallel) -> None:
1 FullyShardedDataParallel title
127 class FullyShardedDataParallel(nn.Module, _FSDPState): class