zhifu gao
2023-02-09 de0ecb446fa429d210397949694d8d9ad6d66112
funasr/bin/asr_inference_paraformer.py
@@ -181,7 +181,7 @@
        self.nbest = nbest
        self.frontend = frontend
        self.encoder_downsampling_factor = 1
        if asr_train_args.encoder_conf["input_layer"] == "conv2d":
        if asr_train_args.encoder == "data2vec_encoder" or asr_train_args.encoder_conf["input_layer"] == "conv2d":
            self.encoder_downsampling_factor = 4
    @torch.no_grad()