singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "Wang Ji (JIRA)" <j...@apache.org>
Subject [jira] [Created] (SINGA-267) Add spatial mode in batch normalization layer
Date Fri, 21 Oct 2016 14:09:58 GMT
Wang Ji created SINGA-267:
-----------------------------

             Summary: Add spatial mode in batch normalization layer 
                 Key: SINGA-267
                 URL: https://issues.apache.org/jira/browse/SINGA-267
             Project: Singa
          Issue Type: Improvement
            Reporter: Wang Ji


Add spatial mode in batch normalization layer in C++ implementation, which corresponds to
CUDNN_BATCHNORM_SPATIAL in CuDNN. 
Also add logics to automatically detect proper modes in batch normalization layer, i.e., if
input is 2D tensor then batchnorm layer chooses PER_ACTIVATION mode, if input is 4D tensor
then batchnorm layer chooses SPATIAL mode.



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

Mime
View raw message