如何init神经网络参数
如何init神经网络参数调用函数语句如何定义weights_init这个函数调用函数语句如果我们要init神经网络的参数,我们可以写这样一条语句:net.apply(weights_init)如何定义weights_init这个函数def weights_init(m):classname = m.class.nameif classname.find(‘Conv’) != -1:nn.init.normal_(m.weight.data, 0.0, 0.02)elif classname
原创
2020-05-23 22:59:29 ·
317 阅读 ·
0 评论