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()
def weight_init(self, mean=0.0, std=0.02): for m in self.named_parameters(): weights_init_normal(m, mean=mean, std=std)
def weight_init(self, mean=0.0, std=0.001): for m in self.modules(): utils.weights_init_normal(m, mean=mean, std=std)