From c1f5bc2e4f9ba4f0ff10b96dca115b55c2e800de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= Date: Thu, 9 Mar 2023 22:33:24 +0800 Subject: [PATCH] modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch --- funasr/models/encoder/resnet34_encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/encoder/resnet34_encoder.py b/funasr/models/encoder/resnet34_encoder.py index 683001bea..7d7179a00 100644 --- a/funasr/models/encoder/resnet34_encoder.py +++ b/funasr/models/encoder/resnet34_encoder.py @@ -831,7 +831,7 @@ class ResNet34SpL2RegDiar(ResNet34_SP_L2Reg): name, data_tf.size(), name_tf, var_dict_tf[name_tf].shape )) else: - var_dict_torch_update[name] = torch.Tensor(map_dict[name]).type(torch.int64).to("cpu") + var_dict_torch_update[name] = torch.from_numpy(np.array(map_dict[name])).type(torch.int64).to("cpu") logging.info("torch tensor: {}, manually assigning to: {}".format( name, map_dict[name] ))