Searched defs:wrap_fsdp (Results 1 – 3 of 3) sorted by relevance
29 def __init__(self, wrap_fsdp): argument46 def _dist_train(self, wrap_fsdp): argument
81 wrap_fsdp=False, argument
97 wrap_fsdp, argument942 def _create_module(wrap_fsdp=True): argument