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)