| 196 | std::cerr << msg.str(); |
| 197 | } |
| 198 | void 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 | |
| 208 | std::unique_ptr<callback_base> |
| 209 | build_debug_callback_from_pbuf(const google::protobuf::Message& proto_msg, |
no test coverage detected