nn.init.kaiming_normal_()
时间: 2024-04-20 13:25:59 浏览: 158
对Pytorch神经网络初始化kaiming分布详解
5星 · 资源好评率100%
nn.init.kaiming_normal_ 是 PyTorch 中的一种权重初始化方法,它使用 "Kaiming normal" 的方式初始化权重。这种方法是在 "Delving deep into rectifiers: Surpassing human-level performance on ImageNet classification" 一文中提出的。它通常用于深度学习中的卷积层。
阅读全文