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

Method on_batch_end

src/callbacks/debug.cpp:121–130  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

119 }
120}
121void debug::on_batch_end(model* m)
122{
123 const auto& c = m->get_execution_context();
124 if (m_modes.empty() || m_modes.count(c.get_execution_mode()) > 0) {
125 std::stringstream msg;
126 msg << rank_string(*m->get_comm()) << ": "
127 << "ending " << batch_step_string(*m) << std::endl;
128 std::cerr << msg.str();
129 }
130}
131void debug::on_batch_evaluate_begin(model* m) { on_batch_begin(m); }
132void debug::on_batch_evaluate_end(model* m) { on_batch_end(m); }
133

Callers 2

do_batch_end_cbsMethod · 0.45
do_batch_end_cbsMethod · 0.45

Calls 6

rank_stringFunction · 0.85
batch_step_stringFunction · 0.85
countMethod · 0.80
emptyMethod · 0.45
get_execution_modeMethod · 0.45
get_commMethod · 0.45

Tested by

no test coverage detected