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)