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] [Closed] (SINGA-267) Add spatial mode in batch normalization layer
Date Fri, 09 Dec 2016 13:37:59 GMT

     [ https://issues.apache.org/jira/browse/SINGA-267?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]

Wang Ji closed SINGA-267.
-------------------------
    Resolution: Fixed
      Assignee: Wang Ji

> 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
>            Assignee: 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