Update ParameterUpdaterHook.cpp

8 years ago · 885275ee77
parent a3ada68ae0
commit 885275ee77
1 changed files with 2 additions and 4 deletions
--- a/paddle/parameter/ParameterUpdaterHook.cpp
+++ b/paddle/parameter/ParameterUpdaterHook.cpp
@ -70,10 +70,8 @@ public:
    for (size_t i = 0; i < para->getSize(); i++)
      param.push_back(std::make_pair(fabs(vecCpu->getData()[i]), i));

-    std::partial_sort(param.begin(),
-                      param.begin() + nonZeroNum,
-                      param.end(),
-                      sortPairAscend);
+    std::partial_sort(
+        param.begin(), param.begin() + nonZeroNum, param.end(), sortPairAscend);
    for (size_t i = 0; i < nonZeroNum; i++) dataPtr[param[i].second] = 1.0;

    // Currently just use a mask vector for hack.