1. 首页
  2. 编程语言
  3. Python
  4. pytorch  网络参数 weight bias 初始化详解

pytorch  网络参数 weight bias 初始化详解

上传者: 2021-05-02 10:26:05上传 PDF文件 88.18 KB 热度 8次

权重初始化对于训练神经网络至关重要,好的初始化权重可以有效的避免梯度消失等问题的发生。在pytorch的使用过程中有几种权重初始化的方法供大家参考。编写好weights_init函数后,可以使用模型的apply方法对模型进行权重初始化。第2节中的内容可以满足一般的参数分组需求,此部分可以满足更个性化的分组需求。使用此方法的前提是,须按下文1,2中的方式定义模型,或者利用Sequential+OrderedDict定义模型。参数分组的区别,其实对应了模型构造时的区别。

下载地址
用户评论