From bc6da7a12379b064111b1fa7fd6c43eac4fed628 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Wed, 17 Mar 2021 08:32:18 +0000
Subject: [PATCH] not hack size since it exists

---
 deepspeech/modules/__init__.py | 14 +++++++-------
 deepspeech/modules/mask.py     |  4 ++--
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/deepspeech/modules/__init__.py b/deepspeech/modules/__init__.py
index c8c35c8ba..925848e86 100644
--- a/deepspeech/modules/__init__.py
+++ b/deepspeech/modules/__init__.py
@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
-from typeing import Union
-from typeing import Any
+from typing import Union
+from typing import Any
 
 import paddle
 from paddle import nn
@@ -21,6 +21,7 @@ from paddle.nn import functional as F
 from paddle.nn import initializer as I
 
 logger = logging.getLogger(__name__)
+logger.warn = logging.warning
 
 # TODO(Hui Zhang): remove this hack
 paddle.bool = 'bool'
@@ -52,11 +53,10 @@ def size(xs: paddle.Tensor, *args: int) -> paddle.Tensor:
         return s
 
 
-if not hasattr(paddle.Tensor, 'size'):
-    logger.warn(
-        "override size of paddle.Tensor if exists or register, remove this when fixed!"
-    )
-    paddle.Tensor.size = size
+# logger.warn(
+#     "override size of paddle.Tensor if exists or register, remove this when fixed!"
+# )
+# paddle.Tensor.size = size
 
 
 def masked_fill(xs: paddle.Tensor,
diff --git a/deepspeech/modules/mask.py b/deepspeech/modules/mask.py
index 7d30f060d..5749c353c 100644
--- a/deepspeech/modules/mask.py
+++ b/deepspeech/modules/mask.py
@@ -272,6 +272,6 @@ def mask_finished_preds(pred: paddle.Tensor, flag: paddle.Tensor,
     Returns:
         paddle.Tensor: (batch_size * beam_size).
     """
-    beam_size = pred.size(-1)
-    finished = flag.repeat([1, beam_size])
+    beam_size = pred.shape[-1]
+    finished = flag.repeat(1, beam_size)
     return pred.masked_fill_(finished, eos)