Repository navigation
Weird code in batch normalization #1096
Description
New Issue Checklist
- [v] I have read the Contribution Guidelines
- [v] I searched for existing GitHub issues
Issue Description
if self.axes is None:
self.axes = [i for i in range(len(inputs.shape)) if i != self.channel_axis]If I understood correctly, the above code is calculating axes to get means and variances for the normalization.
But it looks weird to me because self.channel_axis can be -1. Shouldn't this change like this or something similar?
if self.axes is None:
self.axes = [i for i in range(len(inputs.shape)) if i != self.channel_axis]
if self.channel_axis == -1:
self.axes = self.axes[-1]
Activity
Laicheng0830 commentedon Oct 10, 2020
Thanks! there is a problem with this code, we will fix it immediately.