From 7a207808bc89935a8494c3a9da1ae7fea1c37fd1 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 06 七月 2023 19:13:00 +0800
Subject: [PATCH] np fix bug
---
funasr/modules/nets_utils.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/funasr/modules/nets_utils.py b/funasr/modules/nets_utils.py
index 397a5c4..55c5768 100644
--- a/funasr/modules/nets_utils.py
+++ b/funasr/modules/nets_utils.py
@@ -407,7 +407,7 @@
elif mode == "mt" and arch == "rnn":
# +1 means input (+1) and layers outputs (train_args.elayer)
- subsample = np.ones(train_args.elayers + 1, dtype=np.int)
+ subsample = np.ones(train_args.elayers + 1, dtype=np.int32)
logging.warning("Subsampling is not performed for machine translation.")
logging.info("subsample: " + " ".join([str(x) for x in subsample]))
return subsample
@@ -417,7 +417,7 @@
or (mode == "mt" and arch == "rnn")
or (mode == "st" and arch == "rnn")
):
- subsample = np.ones(train_args.elayers + 1, dtype=np.int)
+ subsample = np.ones(train_args.elayers + 1, dtype=np.int32)
if train_args.etype.endswith("p") and not train_args.etype.startswith("vgg"):
ss = train_args.subsample.split("_")
for j in range(min(train_args.elayers + 1, len(ss))):
@@ -432,7 +432,7 @@
elif mode == "asr" and arch == "rnn_mix":
subsample = np.ones(
- train_args.elayers_sd + train_args.elayers + 1, dtype=np.int
+ train_args.elayers_sd + train_args.elayers + 1, dtype=np.int32
)
if train_args.etype.endswith("p") and not train_args.etype.startswith("vgg"):
ss = train_args.subsample.split("_")
@@ -451,7 +451,7 @@
elif mode == "asr" and arch == "rnn_mulenc":
subsample_list = []
for idx in range(train_args.num_encs):
- subsample = np.ones(train_args.elayers[idx] + 1, dtype=np.int)
+ subsample = np.ones(train_args.elayers[idx] + 1, dtype=np.int32)
if train_args.etype[idx].endswith("p") and not train_args.etype[
idx
].startswith("vgg"):
--
Gitblit v1.9.1