speech_asr il y a 3 ans
Parent
commit
ab6d93b4eb
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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)