阅读背景:

Pytorch 参数初始化以及Xavier初始化

来源:互联网 
    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)    def _initialize_weights(self):
        #



你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: