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

Method on_optimize_begin

src/callbacks/debug.cpp:189–197  ·  view source on GitHub ↗

Status updates for optimization step

Source from the content-addressed store, hash-verified

187
188// Status updates for optimization step
189void debug::on_optimize_begin(model* m, weights* w)
190{
191 auto& dtw = dynamic_cast<data_type_weights<DataType>&>(*w);
192 std::stringstream msg;
193 msg << rank_string(*m->get_comm()) << ": " << weights_string(dtw)
194 << " is starting optimization step for " << batch_step_string(*m)
195 << std::endl;
196 std::cerr << msg.str();
197}
198void debug::on_optimize_end(model* m, weights* w)
199{
200 auto& dtw = dynamic_cast<data_type_weights<DataType>&>(*w);

Callers 2

Calls 4

rank_stringFunction · 0.85
weights_stringFunction · 0.85
batch_step_stringFunction · 0.85
get_commMethod · 0.45

Tested by

no test coverage detected