float16 has a max range of +- 65504 and will overflow to +- Inf outside of this range.
It’s thus expected that nn.MSELoss will overflow via (0.03 - 60000)**2 ~= 3.6e9
float16 has a max range of +- 65504 and will overflow to +- Inf outside of this range.
It’s thus expected that nn.MSELoss will overflow via (0.03 - 60000)**2 ~= 3.6e9