resnest fix (#276)

Oneflow-Inc · Oct 26, 2023 · adab8ca · adab8ca
1 parent c12a04b
commit adab8ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flowvision/models/resnest.py b/flowvision/models/resnest.py
@@ -355,7 +355,7 @@ def __init__(
                 m.weight.data.normal_(0, math.sqrt(2.0 / n))
             elif isinstance(m, norm_layer):
                 m.weight.data.fill_(1)
-                m.bias.data.zeros_()
+                m.bias.data.zero_()
 
     def _make_layer(
         self,