MCPcopy
hub / github.com/huggingface/transformers / opt_load_hook

Method opt_load_hook

src/transformers/trainer.py:3659–3660  ·  view source on GitHub ↗
(mod, opt)

Source from the content-addressed store, hash-verified

3657 if is_sagemaker_mp_enabled():
3658
3659 def opt_load_hook(mod, opt):
3660 opt.load_state_dict(smp.load(os.path.join(checkpoint, OPTIMIZER_NAME), partial=True))
3661
3662 self.model_wrapped.register_post_step_hook(opt_load_hook)
3663 else:

Callers

nothing calls this directly

Calls 2

joinMethod · 0.80
load_state_dictMethod · 0.45

Tested by

no test coverage detected