如何init神经网络参数

调用函数语句

如果我们要init神经网络的参数，我们可以写这样一条语句：
net.apply(weights_init)

如何定义weights_init这个函数

def weights_init(m):
classname = m.class.name
if classname.find(‘Conv’) != -1:
nn.init.normal_(m.weight.data, 0.0, 0.02)
elif classname.find(‘BatchNorm’) != -1:
nn.init.normal_(m.weight.data, 1.0, 0.02)
nn.init.constant_(m.bias.data, 0)

就定义成这个样子，具体需求，视情况而定