m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

"formatter in docker"

Browse Source
gangliao-patch-1
dongzhihong 8 years ago
parent 03884f0fd9
commit a166e5291d
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File

7
paddle/optimizer/parameter_optimizer.cc
Unescape View File

@ -26,10 +26,11 @@ ParameterOptimizer *ParameterOptimizer::Create(const std::string &config_proto,
LOG(WARNING) << " have not select any LrPolicy. use ConstLr in default"; LOG(WARNING) << " have not select any LrPolicy. use ConstLr in default";
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,
lr, lr,

Write Preview
Loading…
Cancel
Save