Searched defs:FSDPTestModel (Results 1 – 1 of 1) sorted by relevance
85 class FSDPTestModel(nn.Module, ABC): class652 module_class: Type[FSDPTestModel],1286 model_class: Type[FSDPTestModel],