|
|
|
@ -27,9 +27,9 @@ ParameterOptimizer *ParameterOptimizer::Create(const std::string &config_proto,
|
|
|
|
|
return new ConstLr(0.1);
|
|
|
|
|
};
|
|
|
|
|
LrPolicy *lr = select_lr_policy(config);
|
|
|
|
|
auto select_optimizer =
|
|
|
|
|
[=](Tensor *parameter,
|
|
|
|
|
const OptimizerConfig &config) -> ParameterOptimizer * {
|
|
|
|
|
auto select_optimizer = [=](
|
|
|
|
|
Tensor *parameter,
|
|
|
|
|
const OptimizerConfig &config) -> ParameterOptimizer * {
|
|
|
|
|
if (config.optimizer() == OptimizerConfig::SGD) {
|
|
|
|
|
return new SGDOptimizer(parameter,
|
|
|
|
|
lr,
|
|
|
|
|