Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 6ebfc72b 编写于 作者: H helinwang 提交者： GitHub
浏览文件
操作

Merge pull request #5104 from helinwang/momentum 

Fix pserver SGD with momemtum checkpoint recover.
上级 a3842494 01df52aa
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
paddle/optimizer/sgd_optimizer.cc
浏览文件 @ 6ebfc72b
...@@ -44,7 +44,7 @@ void SGDOptimizer::DeserializeState(const std::string &str) { ...@@ -44,7 +44,7 @@ void SGDOptimizer::DeserializeState(const std::string &str) {
this->lr_policy_->DeserializeState(lr_state.SerializeAsString()); this->lr_policy_->DeserializeState(lr_state.SerializeAsString());
num_sample_passed_ = state.num_sample_passed(); num_sample_passed_ = state.num_sample_passed();
ProtoToTensor(state.parameter(), parameter_); ProtoToTensor(state.parameter(), parameter_);
if (momentum_ != 0.0) ProtoToTensor(state.parameter(), momentums_); if (momentum_ != 0.0) ProtoToTensor(state.momentums(), momentums_);
} }
} // namespace optimizer } // namespace optimizer
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或