pytorch
ac72881f - Fix a numerical issue of CUDA channels-last SyncBatchNorm (#57077)

Commit
3 years ago
Fix a numerical issue of CUDA channels-last SyncBatchNorm (#57077) Summary: Fix a numerical issue of CUDA channels-last SyncBatchNorm The added test is a repro for the numerical issue. Thanks for the help from jjsjann123 who identified the root cause. Since pytorch SBN channels-last code was migrated from [nvidia/apex](https://github.com/nvidia/apex), apex SBN channels-last also has this issue. We will submit a fix there soon. Pull Request resolved: https://github.com/pytorch/pytorch/pull/57077 Reviewed By: mruberry Differential Revision: D28107672 Pulled By: ngimel fbshipit-source-id: 0c80e79ddb48891058414ad8a9bedd80f0f7f8df
Author
Parents
Loading