Closed xuanyuanminzheng closed 2 months ago
nan occasionally occurs in the global mean and variance of batchnorm, which may be caused by the axpy kernel.
Thanks for your contribution!
nan occasionally occurs in the global mean and variance of batchnorm, which may be caused by the axpy kernel.