chrishkchris edited a comment on issue #554: [bug]batchnorm running mean var not updated URL: https://github.com/apache/singa/pull/554#issuecomment-546714563 I think this is pass by reference (the running mean and variance singa tensors' blocks are fed to the function cudnnBatchNormalizationForwardTraining), so there is no need to return the running mean and variance singa tensors (because the _BatchNorm2d already have them)?
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
