From 0d1995c82c2ad6caeecbcbc4c35607d6ea28f99e Mon Sep 17 00:00:00 2001 From: Mu Li Date: Fri, 10 Apr 2015 12:32:48 -0400 Subject: [PATCH] add key_caching in pull --- src/app/linear_method/async_sgd.h | 1 + 1 file changed, 1 insertion(+) diff --git a/src/app/linear_method/async_sgd.h b/src/app/linear_method/async_sgd.h index 668253e..a1e95c8 100644 --- a/src/app/linear_method/async_sgd.h +++ b/src/app/linear_method/async_sgd.h @@ -228,6 +228,7 @@ class AsyncSGDWorker : public ISGDCompNode, public LinearMethod { MessagePtr msg(new Message(kServerGroup)); msg->setKey(key); msg->task.set_key_channel(id); + msg->addFilter(FilterConfig::KEY_CACHING) msg->fin_handle = [this, id]() { computeGradient(id); }; model_.key(id) = key; model_.pull(msg);