zhuchen03 / gradinit

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

small fixes #4

Closed Xrenya closed 2 years ago

Xrenya commented 2 years ago

just fix model name and check dir to run the model without any problem