pytorch code

fix https://github.com/WongKinYiu/yolov7/issues/855
2022-09-28 17:01:55 +08:00 · 2022-09-28 17:01:55 +08:00 · 88e9436bec
parent d05e8ba88d
commit 88e9436bec
1 changed files with 1 additions and 1 deletions
--- 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