Skip to content

resolve std computation nan#108

Open
LVeefkind wants to merge 1 commit intoQUVA-Lab:masterfrom
LVeefkind:fix_batchnorm
Open

resolve std computation nan#108
LVeefkind wants to merge 1 commit intoQUVA-Lab:masterfrom
LVeefkind:fix_batchnorm

Conversation

@LVeefkind
Copy link
Copy Markdown
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