Skip to content

Conversation

@LVeefkind
Copy link
Collaborator

Prevents NaN std computation in the _estimate_stats() function for the _IIDBatchNorm. This prevents issues in the forward() call when the inputs are all the same.

Signed-off-by: LVeefkind <larsveefkind@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant