Function load_inference_model

src/utils/lbann_library.cpp:55–70

Loads a model from checkpoint and sets up model for inference

// Loads a model from checkpoint and sets up model for inference
std::unique_ptr<model> load_inference_model(lbann_comm* lc,
                                            std::string cp_dir,
                                            int mbs,
                                            std::vector<El::Int> input_dims,
                                            std::vector<El::Int> output_dims)
{
  persist p;
  p.open_restart(cp_dir.c_str());
  auto m = std::make_unique<model>(lc, nullptr, nullptr);
  m->load_from_checkpoint_shared(p);
  p.close_restart();

  m->setup(mbs, get_trainer().get_grids());

  return m;
}

/// Split the MPI communicator into trainers
/// Return the