Przeglądaj źródła

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

志浩 3 lat temu
rodzic
commit
f609d6ce50

+ 1 - 0
egs/callhome/diarization/sond/sond.yaml

@@ -2667,6 +2667,7 @@ encoder_conf:
     # pass by model, equal to feature dim
     # input_size: 80
     pooling_type: "window_shift"
+    batchnorm_momentum: 0.01
     pool_size: 20
     stride: 1
     tf2torch_tensor_name_prefix_torch: encoder

+ 1 - 0
egs/callhome/diarization/sond/sond_fbank.yaml

@@ -2666,6 +2666,7 @@ encoder: resnet34_sp_l2reg
 encoder_conf:
     # pass by model, equal to feature dim
     # input_size: 80
+    batchnorm_momentum: 0.01
     pooling_type: "window_shift"
     pool_size: 20
     stride: 1