main code

fix #855
WongKinYiu · Sep 28, 2022 · 8035ee6 · 8035ee6
1 parent 55b90e1
commit 8035ee6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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