You are viewing a plain text version of this content. The canonical link for it is here.
Posted to dev@singa.apache.org by "wangwei (JIRA)" <ji...@apache.org> on 2016/10/16 07:52:21 UTC

[jira] [Created] (SINGA-262) Change the dimension order of the weight matrix for the dense layer

wangwei created SINGA-262:
-----------------------------

             Summary: Change the dimension order of the weight matrix for the dense layer
                 Key: SINGA-262
                 URL: https://issues.apache.org/jira/browse/SINGA-262
             Project: Singa
          Issue Type: Improvement
            Reporter: wangwei
            Assignee: ZHAOJING


The dimension of the weight matrix of the Dense layer in V1.0 is (num_input, num_output), whereas it is (num_output, num_input) in Caffe and Torch. To make it easy to convert caffe models, we can change the dimension order. 

To be compatible with previous checkpoint files which were generated with (num_input, num_output), we need to change the version ID to (1.0.1) and add it into the checkpoint file via SINGA-261. The net load() function should be updated to convert the weight matrix according to the version ID: transpose the weight matrix if it was dumped from v1.0.0; do nothing for >=v1.0.1.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)