Просмотр исходного кода

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

志浩 3 лет назад
Родитель
Сommit
c3ca7d963e
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      funasr/models/encoder/resnet34_encoder.py

+ 1 - 1
funasr/models/encoder/resnet34_encoder.py

@@ -740,7 +740,7 @@ class ResNet34SpL2RegDiar(ResNet34_SP_L2Reg):
                 "{}.resnet{}_dense.weight".format(tensor_name_prefix_torch, layer_idx):
                     {"name": "{}/resnet{}_dense/kernel".format(tensor_name_prefix_tf, layer_idx),
                      "squeeze": None,
-                     "transpose": (1, 0),
+                     "transpose": (2, 1, 0) if layer_idx == 0 else (1, 0),
                      },
                 "{}.resnet{}_dense.bias".format(tensor_name_prefix_torch, layer_idx):
                     {"name": "{}/resnet{}_dense/bias".format(tensor_name_prefix_tf, layer_idx),