Fix optimizer's initialization

utcs-scea · May 31, 2024 · d504249 · d504249
1 parent 3787fd3
commit d504249
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/workflows/vertex_classification/include/pando-wf1/optimizer.hpp b/workflows/vertex_classification/include/pando-wf1/optimizer.hpp
@@ -60,7 +60,8 @@ class AdamOptimizer {
     galois::doAll(
         Tpl{this->secondMoments_, this->beta1Power_, this->beta2Power_, numTrainableLayers,
             trainableLayerSizes, this->config_},
-        this->firstMoments_, +[](Tpl tpl, pando::GlobalRef<pando::Array<pando::Array<GNNFloat>>> fmRef) {
+        this->firstMoments_,
+        +[](Tpl tpl, pando::GlobalRef<pando::Array<pando::Array<GNNFloat>>> fmRef) {
           std::uint32_t host = pando::getCurrentPlace().node.id;
 
           pando::GlobalRef<pando::Array<pando::Array<GNNFloat>>> smRef = *fmap(tpl.sm, get, host);
@@ -91,7 +92,6 @@ class AdamOptimizer {
             pando::Array<GNNFloat> ifm = ifmRef;
             pando::Array<GNNFloat> ism = ismRef;
 
-
             galois::doAll(
                 ifm, +[](pando::GlobalRef<GNNFloat> v) {
                   v = 0;