背景:

  • BP在产生的过程中:会产生梯度消失/梯度爆炸。 为了阻止这种情况发生

 

优点与缺点:

  • 优点:简单粗暴
  • 缺点:很难找到满意的阈值

 

代码:

torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=20)

 

 

 

 

参考:https://www.cnblogs.com/lindaxin/p/7998196.html

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐