SINGA-140 patch

apache · Mar 30, 2016 · 3b87183 · 3b87183
1 parent 259422c
commit 3b87183
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);
       }