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

志浩

2023-03-09 c1f5bc2e4f9ba4f0ff10b96dca115b55c2e800de

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

 funasr/models/encoder/resnet34_encoder.py

@@ -831,7 +831,7 @@
                            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]
                        ))