nichongjia-2007
2023-05-11 0a51365f8af7d034c5fddf8fa7c9e810f3296d17
funasr/tasks/asr.py
@@ -103,7 +103,7 @@
    name="specaug",
    classes=dict(
        specaug=SpecAug,
        specaug_lfr=SpecAugLFR,
        specaug_lfr=FSpecAugLR,
    ),
    type_check=AbsSpecAug,
    default=None,
@@ -1684,7 +1684,7 @@
        # 7. Build model
        if encoder.unified_model_training:
        if hasattr(encoder, 'unified_model_training') and encoder.unified_model_training:
            model = UnifiedTransducerModel(
                vocab_size=vocab_size,
                token_list=token_list,