5.3 神经网络参数初始化

一、内置随机化 init函数

二、内置常数化 init函数

三、不同层 or 块进行不同的init

四、自定义init函数

解释

1、apply(my_init)之后,神经网络会将每一层的权重和偏置分别放入my_init()函数并将结果赋予它们自身

2、my_init()函数接收每一层的权重和偏置,并采用torch的随机初始化(这是我随便写的,可以根据需求改写)

3、net.weight / net.bias的数据类型是torch.nn.parameter.Parameter,需要用.data将其转换为torch.tensor才能进行赋值

4、net.weight / net.bias的数据类型是torch.nn.parameter.Parameter,也可以用torch.nn.Parameter直接进行初始化,而不需要转换

演示

五、参数直接赋值 & 不同层 参数绑定

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值