示例#1
0
    def weight_init(self):
        for m in self.modules():
            utils.weights_init_kaming(m)

        if self.config['input_conv_sigma']:
            for m in self.input_conv.modules():
                utils.weights_init_normal(m,
                                          std=self.config['input_conv_sigma'])

        self.ImageReconstruction.init_tconv()
示例#2
0
 def weight_init(self, mean=0.0, std=0.02):
     for m in self.named_parameters():
         weights_init_normal(m, mean=mean, std=std)
示例#3
0
 def weight_init(self, mean=0.0, std=0.001):
     for m in self.modules():
         utils.weights_init_normal(m, mean=mean, std=std)