Fix pserver SGD with momemtum checkpoing recover.

01df52aa · Helin Wang · 0760043d · 01df52aa
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/optimizer/sgd_optimizer.cc paddle/optimizer/sgd_optimizer.cc +1 -1

未找到文件。
--- 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