From 01df52aaad5fb629da69a3dc6e18aceca340d8cb Mon Sep 17 00:00:00 2001 From: Helin Wang Date: Wed, 25 Oct 2017 16:58:59 -0700 Subject: [PATCH] Fix pserver SGD with momemtum checkpoing recover. --- paddle/optimizer/sgd_optimizer.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/optimizer/sgd_optimizer.cc b/paddle/optimizer/sgd_optimizer.cc index bf2540ecb09..1090419083c 100644 --- a/paddle/optimizer/sgd_optimizer.cc +++ b/paddle/optimizer/sgd_optimizer.cc @@ -44,7 +44,7 @@ void SGDOptimizer::DeserializeState(const std::string &str) { this->lr_policy_->DeserializeState(lr_state.SerializeAsString()); num_sample_passed_ = state.num_sample_passed(); ProtoToTensor(state.parameter(), parameter_); - if (momentum_ != 0.0) ProtoToTensor(state.parameter(), momentums_); + if (momentum_ != 0.0) ProtoToTensor(state.momentums(), momentums_); } } // namespace optimizer -- GitLab