SINGA-140

raunaqabhyankar · raunaqabhyankar · commit ac36b34621c1 · 2016-03-30T15:56:09.000+05:30
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);
       }

Original file line number	Diff line number	Diff line change
`@@ -297,7 +297,7 @@ int Worker::Update(int step, Param* param) {`
`297`	`297`	`int Worker::CollectAll(int step, NeuralNet* net) {`
`298`	`298`	`auto& layers = net->layers();`
`299`	`299`	`for (auto& layer : layers) {`
`300`		`- if (layer->partition_id() == id_) {`
	`300`	`+ if (layer->partition_id() == id_ && layer->unroll_index() == 0) {`
`301`	`301`	`for (Param* p : layer->GetParams()) {`
`302`	`302`	`Collect(step, p);`
`303`	`303`	`}`