Searched defs:set_requires_gradient_sync (Results 1 – 2 of 2) sorted by relevance
163 def set_requires_gradient_sync(self, requires_gradient_sync: bool) -> None: member in DDP
226 def set_requires_gradient_sync( member in FSDPModule