square / pysurvival

Open source package for Survival Analysis modeling
https://www.pysurvival.io/
Apache License 2.0
350 stars 106 forks source link

added clip_grad_norm #11

Open kush99993s opened 5 years ago

kush99993s commented 5 years ago

Hello @steph-likes-git,

I wanted to add clipping of gradient to make sure that we are not exploding our gradient especially for ReLU, Identity, or similar activation functions where gradient can explode very easily.