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

Method on_optimize_end

src/callbacks/debug.cpp:198–206  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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);
201 std::stringstream msg;
202 msg << rank_string(*m->get_comm()) << ": " << weights_string(dtw)
203 << " is ending optimization step for " << batch_step_string(*m)
204 << std::endl;
205 std::cerr << msg.str();
206}
207
208std::unique_ptr<callback_base>
209build_debug_callback_from_pbuf(const google::protobuf::Message& proto_msg,

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