Zheng Kaiping created SINGA-140:
-----------------------------------
Summary: A bug in CollectAll() function when layers are unrolled
in neuralnet
Key: SINGA-140
URL: https://issues.apache.org/jira/browse/SINGA-140
Project: Singa
Issue Type: Bug
Reporter: Zheng Kaiping
In SINGA_HOME/src/work.cc, in “int Worker::CollectAll(int step, NeuralNet*
net){}” function, the layers which are unrolled (except for the first one)
should not collect parameters, due to parameter sharing.
Previous:
if (layer->partition_id() == id_)
Possible changes:
if (layer->partition_id() == id_ && layer->unroll_index() == 0)
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
