提交 03884f0f 编写于 作者: D dzhwinter 提交者: GitHub

formatter

上级 72b6b26f
...@@ -27,8 +27,8 @@ ParameterOptimizer *ParameterOptimizer::Create(const std::string &config_proto, ...@@ -27,8 +27,8 @@ ParameterOptimizer *ParameterOptimizer::Create(const std::string &config_proto,
return new ConstLr(0.1); return new ConstLr(0.1);
}; };
LrPolicy *lr = select_lr_policy(config); LrPolicy *lr = select_lr_policy(config);
auto select_optimizer = auto select_optimizer = [=](
[=](Tensor *parameter, Tensor *parameter,
const OptimizerConfig &config) -> ParameterOptimizer * { const OptimizerConfig &config) -> ParameterOptimizer * {
if (config.optimizer() == OptimizerConfig::SGD) { if (config.optimizer() == OptimizerConfig::SGD) {
return new SGDOptimizer(parameter, return new SGDOptimizer(parameter,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册