def _initialize_weights(self):
# print(self.modules())
for m in self.modules():
print(m)
if isinstance(m, nn.Linear):
# print(m.weight.data.type())
# input()
# m.weight.data.fill_(1.0)
init.xavier_uniform_(m.weight, gain=1)
print(m.weight)
可以再网络的类中定义初始化初始化函数
通过net._initialize_weights()进行初始化
Xavier的具体讲解可以参考