fix prod (#3763)

40436c83 · Wenyu · GitHub · b42baeb8 · 40436c83
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/initializer.py ppdet/modeling/initializer.py +1 -1

未找到文件。
--- a/ppdet/modeling/initializer.py
+++ b/ppdet/modeling/initializer.py
@@ -267,7 +267,7 @@ def linear_init_(module):


 def conv_init_(module):
-    bound = 1 / math.sqrt(math.prod(module.weight.shape[1:]))
+    bound = 1 / np.sqrt(np.prod(module.weight.shape[1:]))
    uniform_(module.weight, -bound, bound)
    uniform_(module.bias, -bound, bound)