Searched refs:FSDPTestModel (Results 1 – 1 of 1) sorted by relevance
85 class FSDPTestModel(nn.Module, ABC): class236 class TransformerWithSharedParams(FSDPTestModel):384 class NestedWrappedModule(FSDPTestModel):608 class ModuleWithDelay(FSDPTestModel):652 module_class: Type[FSDPTestModel], argument1286 model_class: Type[FSDPTestModel], argument