From 81d54c3ee21a9c90c5cb108a8b739ba36c70625c Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Tue, 15 Jun 2021 16:02:02 +0800
Subject: [PATCH] fix prod (#3377)

---
 ppdet/modeling/initializer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppdet/modeling/initializer.py b/ppdet/modeling/initializer.py
index 96f5eee2f..ce5dea95a 100644
--- a/ppdet/modeling/initializer.py
+++ b/ppdet/modeling/initializer.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import math
+import numpy as np
 
 import paddle
 import paddle.nn as nn
@@ -136,7 +137,7 @@ def _calculate_fan_in_and_fan_out(tensor, reverse=False):
 
     receptive_field_size = 1
     if tensor.ndim > 2:
-        receptive_field_size = math.prod(tensor.shape[2:])
+        receptive_field_size = np.prod(tensor.shape[2:])
 
     fan_in = num_input_fmaps * receptive_field_size
     fan_out = num_output_fmaps * receptive_field_size
-- 
GitLab