diff --git a/models/common.py b/models/common.py index 111af708de..edb5edc9fe 100644 --- a/models/common.py +++ b/models/common.py @@ -444,7 +444,7 @@ def forward(self, x): class ImplicitM(nn.Module): - def __init__(self, channel, mean=0., std=.02): + def __init__(self, channel, mean=1., std=.02): super(ImplicitM, self).__init__() self.channel = channel self.mean = mean