defweight_init(self):for m in self._modules:ifisinstance(m, nn.Linear)orisinstance(m, nn.Conv2d):#判断当前网络结构是否为全连接层或者卷积层
m.weight.data.normal_(0.0,0.01)
m.bias.data.zero_()
定义网络结构时,对网络结构参数进行初始化,有时更利于网络的训练收敛 def weight_init(self): for m in self._modules: if isinstance(m, nn.Linear) or isinstance(m, nn.Conv2d): #判断当前网络结构是否为全连接层或者卷积层 m.weight.data.normal_(0.0, 0.01) m.bias