speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

Browse Source
This commit is contained in:
志浩 2023-03-09 22:33:24 +08:00
parent f609d6ce50
commit c1f5bc2e4f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/encoder/resnet34_encoder.py
View File

@ -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]
))