jmwang66
2023-02-06 6bb6af36ac4e3a3bea69b36c7022896e18f9a079
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()