grad become nan in first iter update #1
Description
after my first backword, the grad of shared model parameter become nan, what's the reason?
backword in this place:
https://github.com/brianlan/complex-grad-norm/blob/master/src/gradnorm.py#L90
{{ message }}