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

Method on_forward_prop_end

src/callbacks/debug.cpp:146–156  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

144 }
145}
146void debug::on_forward_prop_end(model* m, Layer* l)
147{
148 const auto& c = m->get_execution_context();
149 if (m_modes.empty() || m_modes.count(c.get_execution_mode()) > 0) {
150 std::stringstream msg;
151 msg << rank_string(*m->get_comm()) << ": " << layer_string(*l)
152 << " is ending forward prop for " << batch_step_string(*m)
153 << std::endl;
154 std::cerr << msg.str();
155 }
156}
157void debug::on_backward_prop_begin(model* m, Layer* l)
158{
159 const auto& c = m->get_execution_context();

Callers

nothing calls this directly

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