gangliao-patch-1
dzhwinter 8 years ago committed by GitHub
parent 72b6b26f5a
commit 03884f0fd9

@ -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,

Loading…
Cancel
Save