speech_asr há 3 anos atrás
pai
commit
ab6d93b4eb
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      funasr/tasks/diar.py

+ 1 - 1
funasr/tasks/diar.py

@@ -823,7 +823,7 @@ class EENDOLADiarTask(AbsTask):
 
         # 2. Encoder
         encoder_class = encoder_choices.get_class(args.encoder)
-        encoder = encoder_class(input_size=input_size, **args.encoder_conf)
+        encoder = encoder_class(**args.encoder_conf)
 
         # 3. EncoderDecoderAttractor
         encoder_decoder_attractor_class = encoder_decoder_attractor_choices.get_class(args.encoder_decoder_attractor)