| 544 | } |
| 545 | |
| 546 | std::unique_ptr<callback_base> build_step_learning_rate_callback_from_pbuf( |
| 547 | const google::protobuf::Message& proto_msg, |
| 548 | const std::shared_ptr<lbann_summary>&) |
| 549 | { |
| 550 | const auto& params = |
| 551 | dynamic_cast<const lbann_data::Callback::CallbackStepLearningRate&>( |
| 552 | proto_msg); |
| 553 | return std::make_unique<step_learning_rate>( |
| 554 | params.step(), |
| 555 | params.amt(), |
| 556 | parse_list<std::string>(params.weights())); |
| 557 | } |
| 558 | |
| 559 | std::unique_ptr<callback_base> build_set_learning_rate_callback_from_pbuf( |
| 560 | const google::protobuf::Message& proto_msg, |