chrishkchris removed a comment on issue #554: [bug]batchnorm running mean var not updated URL: https://github.com/apache/singa/pull/554#issuecomment-546757182 Oh I see. The running mean and var are updated using NOT in place assignment. running_mean = running_mean * bnh.factor + mean * (1 - bnh.factor); If we change it to in place operation, there is no need to return, i.e. running_mean*=bnh.factor; Axpy(1-bnh.factor,mean,running_mean)
---------------------------------------------------------------- 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
