Searched defs:fsdp_wrap (Results 1 – 2 of 2) sorted by relevance
58 def fsdp_wrap(self, fsdp_kwargs): member in Model92 def fsdp_wrap(self, fsdp_kwargs): member in NestedTrunkModel
1510 def __init__(self, fsdp_wrap): argument