| 557 | } |
| 558 | |
| 559 | std::unique_ptr<callback_base> build_set_learning_rate_callback_from_pbuf( |
| 560 | const google::protobuf::Message& proto_msg, |
| 561 | const std::shared_ptr<lbann_summary>&) |
| 562 | { |
| 563 | const auto& params = |
| 564 | dynamic_cast<const lbann_data::Callback::CallbackSetLearningRate&>( |
| 565 | proto_msg); |
| 566 | return std::make_unique<set_learning_rate>( |
| 567 | params.step(), |
| 568 | params.val(), |
| 569 | parse_list<std::string>(params.weights())); |
| 570 | } |
| 571 | |
| 572 | std::unique_ptr<callback_base> build_adaptive_learning_rate_callback_from_pbuf( |
| 573 | const google::protobuf::Message& proto_msg, |