Issue with BatchNorm + Conv that depends on dtype

autograd
5

Hello again,

Yes, this solves the issue.

I’ve changed default colab’s version 1.10.0+cu111 to 1.12.0.dev20220209+cu111 and obtained proper behavior.

Thank you