diff --git a/models/common.py b/models/common.py index 111af70..edb5edc 100644 --- a/models/common.py +++ b/models/common.py @@ -444,7 +444,7 @@ class ImplicitA(nn.Module): 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