
参数初始化kaiming
发布日期:2021-05-06 11:08:17
浏览次数:26
分类:精选文章
本文共 835 字,大约阅读时间需要 2 分钟。
-
前向传播的时候, 每一层的卷积计算结果的方差为1.
-
反向传播的时候, 每一 层的继续往前传的梯度方差为1(因为每层会有两个梯度的计算, 一个用来更新当前层的权重, 一个继续传播, 用于前面层的梯度的计算.)
def weights_init_kaiming(m): classname = m.__class__.__name__ if classname.find('Linear') != -1: nn.init.kaiming_normal_(m.weight, a=0, mode='fan_out') nn.init.constant_(m.bias, 0.0) elif classname.find('Conv') != -1: nn.init.kaiming_normal_(m.weight, a=0, mode='fan_in') if m.bias is not None: nn.init.constant_(m.bias, 0.0) elif classname.find('BatchNorm') != -1: if m.affine: nn.init.constant_(m.weight, 1.0) nn.init.constant_(m.bias, 0.0)def weights_init_classifier(m): classname = m.__class__.__name__ if classname.find('Linear') != -1: nn.init.normal_(m.weight, std=0.001) if m.bias: nn.init.constant_(m.bias, 0.0)
发表评论
最新留言
网站不错 人气很旺了 加油
[***.192.178.218]2025年03月30日 04时58分55秒
关于作者

喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
Linux启动Nginx后访问虚拟机的ip时会出现链接不上的问题
2019-03-04
Vue学习(一)——概述及入门
2019-03-04
Vue学习(四十五)——Element-UI
2019-03-04
Docker教程(二)——Docker的安装及配置加速器
2019-03-04
MyBatis(二)——MyBatis入门实例
2019-03-04
mysql常用函数
2019-03-04
Redis 缓存击穿
2019-03-04
从零带你学习Go语言:21、 Go关键词goroutine和channel
2019-03-04
从零带你学习Go语言:23、 Go指针 、切片、 map、 结构体复习
2019-03-04
从零带你学习Go语言:22、 Go基础 、包、变量和函数、流程控制语句复习
2019-03-04
从零带你学习Go语言:25、 Go并发机制
2019-03-04
【C++系列】C++中的基础语法总结
2019-03-04
广东iOS企业开发者签名有没有好的推荐?
2019-03-04