Searched refs:disable_apex_o2_state_dict_hook (Results 1 – 1 of 1) sorted by relevance
110 def disable_apex_o2_state_dict_hook(model: torch.nn.Module | torch.jit.ScriptFunction): function156 ) as mode_ctx, disable_apex_o2_state_dict_hook(