Fixed a bug in the cuDNN binding that on rare occasions caused NaN outputs from
batch normalization. The running mean and variance need to be initialized when calling cuDNN, even if the averaging factor is 1. I don't think this was the case prior to cuDNN v5, but it certainly is the case now. This patch fixes this.
Showing
Please
register
or
sign in
to comment