MCPcopy Create free account
hub / github.com/LBANN/lbann / reload_model

Method reload_model

src/callbacks/checkpoint.cpp:491–501  ·  view source on GitHub ↗

Reload a model from a Shared/Distributed checkpoint

Source from the content-addressed store, hash-verified

489
490// Reload a model from a Shared/Distributed checkpoint
491bool checkpoint::reload_model(model* m)
492{
493 lbann::utils::grid_manager grid_raii(m->get_comm()->get_trainer_grid());
494 return open_latest_checkpoint(
495 *(m->get_comm()),
496 "Reloading Model " + m->get_name(),
497 get_active_trainer().get_name(),
498 get_active_training_algorithm().get_type(),
499 [m](persist& p_ref) { return m->load_from_checkpoint_shared(p_ref); },
500 [m](persist& p_ref) { return m->load_from_checkpoint_distributed(p_ref); });
501}
502
503// Reload a model from a Shared/Distributed checkpoint
504bool checkpoint::reload_trainer(trainer* t)

Callers

nothing calls this directly

Calls 5

get_commMethod · 0.45
get_nameMethod · 0.45
get_typeMethod · 0.45

Tested by

no test coverage detected