sony / nnabla

Neural Network Libraries
https://nnabla.org/
Apache License 2.0
2.73k stars 334 forks source link

Check consistency among memory layouts in SyncBatchNormalization #1169

Closed TakuyaNarihira closed 1 year ago

TakuyaNarihira commented 1 year ago

We found that SyncBatchNormalizationCuda gave an incorrect outputs in channel last layout. It's a test to check if the problem is fixed in sony/nnabla-ext-cuda#457.

EDIT:

It turned out that gradient values of gamma do not match between memory layouts. We defer the investigation in order to merge the change for the first issue we've found and resolved.