[
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)
