游雁 hace 2 años
padre
commit
dfab707845
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  1. 2 0
      funasr/tasks/asr.py

+ 2 - 0
funasr/tasks/asr.py

@@ -280,6 +280,8 @@ class ASRTask(AbsTask):
         postencoder_choices,
         # --decoder and --decoder_conf
         decoder_choices,
+        # --predictor and --predictor_conf
+        predictor_choices,
     ]
 
     # If you need to modify train() or eval() procedures, change Trainer class here