model.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:ShuffleNet 作者: jaxony 项目源码 文件源码
def init_params(self):
        for m in self.modules():
            if isinstance(m, nn.Conv2d):
                init.kaiming_normal(m.weight, mode='fan_out')
                if m.bias is not None:
                    init.constant(m.bias, 0)
            elif isinstance(m, nn.BatchNorm2d):
                init.constant(m.weight, 1)
                init.constant(m.bias, 0)
            elif isinstance(m, nn.Linear):
                init.normal(m.weight, std=0.001)
                if m.bias is not None:
                    init.constant(m.bias, 0)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号