From ac36b34621c14e366a5c1a70e7edeb575fbcf437 Mon Sep 17 00:00:00 2001
From: Raunaq Abhyankar <raunaq.abhyankar@gmail.com>
Date: Wed, 30 Mar 2016 11:44:41 +0530
Subject: [PATCH] SINGA-140

---
 src/worker.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/worker.cc b/src/worker.cc
index 2afa8b06fe..0bcf556f3f 100644
--- a/src/worker.cc
+++ b/src/worker.cc
@@ -297,7 +297,7 @@ int Worker::Update(int step, Param* param) {
 int Worker::CollectAll(int step, NeuralNet* net) {
   auto& layers = net->layers();
   for (auto& layer : layers) {
-    if (layer->partition_id() == id_) {
+    if (layer->partition_id() == id_ && layer->unroll_index() == 0) {		
       for (Param* p : layer->GetParams()) {
         Collect(step, p);
       }