Merge pull request #5104 from helinwang/momentum

helinwang · web-flow · commit 6ebfc72b7259 · 2017-10-25T17:35:53.000-07:00
Fix pserver SGD with momemtum checkpoint recover.
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ void SGDOptimizer::DeserializeState(const std::string &str) {`
`44`	`44`	`this->lr_policy_->DeserializeState(lr_state.SerializeAsString());`
`45`	`45`	`num_sample_passed_ = state.num_sample_passed();`
`46`	`46`	`ProtoToTensor(state.parameter(), parameter_);`
`47`		`- if (momentum_ != 0.0) ProtoToTensor(state.parameter(), momentums_);`
	`47`	`+ if (momentum_ != 0.0) ProtoToTensor(state.momentums(), momentums_);`
`48`	`48`	`}`
`49`	`49`
`50`	`50`	`} // namespace optimizer`