From d504249e53d542577a6e8ad6300b5f011a330e51 Mon Sep 17 00:00:00 2001 From: Hochan Lee Date: Fri, 31 May 2024 02:24:53 -0500 Subject: [PATCH] Fix optimizer's initialization --- .../vertex_classification/include/pando-wf1/optimizer.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/workflows/vertex_classification/include/pando-wf1/optimizer.hpp b/workflows/vertex_classification/include/pando-wf1/optimizer.hpp index 9155cbcd..68359b6f 100644 --- 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>> fmRef) { + this->firstMoments_, + +[](Tpl tpl, pando::GlobalRef>> fmRef) { std::uint32_t host = pando::getCurrentPlace().node.id; pando::GlobalRef>> smRef = *fmap(tpl.sm, get, host); @@ -91,7 +92,6 @@ class AdamOptimizer { pando::Array ifm = ifmRef; pando::Array ism = ismRef; - galois::doAll( ifm, +[](pando::GlobalRef v) { v = 0;