| 357 | } |
| 358 | |
| 359 | void dump_weights::write_specific_proto(lbann_data::Callback& proto) const |
| 360 | { |
| 361 | auto* msg = proto.mutable_dump_weights(); |
| 362 | msg->set_directory(m_directory); |
| 363 | msg->set_epoch_interval(m_epoch_interval); |
| 364 | msg->set_format(to_string(*m_file_format)); |
| 365 | } |
| 366 | |
| 367 | std::unique_ptr<callback_base> build_dump_weights_callback_from_pbuf( |
| 368 | const google::protobuf::Message& proto_msg, |
nothing calls this directly
no test coverage detected