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

Method on_backward_prop_begin

src/callbacks/debug.cpp:157–167  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

155 }
156}
157void debug::on_backward_prop_begin(model* m, Layer* l)
158{
159 const auto& c = m->get_execution_context();
160 if (m_modes.empty() || m_modes.count(c.get_execution_mode()) > 0) {
161 std::stringstream msg;
162 msg << rank_string(*m->get_comm()) << ": " << layer_string(*l)
163 << " is starting backward prop for " << batch_step_string(*m)
164 << std::endl;
165 std::cerr << msg.str();
166 }
167}
168void debug::on_backward_prop_end(model* m, Layer* l)
169{
170 const auto& c = m->get_execution_context();

Calls 7

rank_stringFunction · 0.85
layer_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