def weights_init(m): if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear):#判断m对象是否属于这些类 nn.init.xavier_uniform(m.weight.data) nn.init.constant(m.bias, 0.1) class Net: def __init(): self.apply(weights_init)#权重初始化
文章来源: https://blog.csdn.net/weixin_38241876/article/details/91958751