zhuchen03 / gradinit

Learning to Initialize Neural Networks for Stable and Efficient Training
136 stars 12 forks source link

Update gradinit_utils.py #3

Closed Xrenya closed 2 years ago

Xrenya commented 2 years ago

I think it is a redundant comparison or maybe it should be something different here?