fix prod (#3762)

3c761465 · Wenyu · GitHub · 8019fa9e · 3c761465
隐藏空白更改
内联并排

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)