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

Method model

src/models/model.cpp:72–85  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

70// =============================================
71
72model::model(lbann_comm* comm,
73 std::unique_ptr<objective_function> obj_fn,
74 std::unique_ptr<lbann_data::Optimizer> default_optimizer_msg)
75 : m_execution_context(nullptr),
76 m_comm(comm),
77 m_default_optimizer_msg(std::move(default_optimizer_msg))
78{
79
80 m_objective_function = std::move(obj_fn);
81 // Default model name
82 static El::Int num_models = 0;
83 m_name = "model" + std::to_string(num_models);
84 num_models++;
85}
86
87model::model() : model(&utils::get_current_comm(), nullptr, nullptr) {}
88

Callers 15

train_stepMethod · 0.80
evalMethod · 0.80
forwardMethod · 0.80
init_data_readersFunction · 0.80
print_parametersFunction · 0.80
save_sessionFunction · 0.80
setup_modelFunction · 0.80
make_modelFunction · 0.80
make_modelFunction · 0.80
make_modelFunction · 0.80

Calls 6

set_modelMethod · 0.80
to_stringFunction · 0.50
copyMethod · 0.45
sizeMethod · 0.45
get_nameMethod · 0.45
getMethod · 0.45

Tested by 4

setup_modelFunction · 0.64
make_modelFunction · 0.64
make_modelFunction · 0.64
make_modelFunction · 0.64