add SetBatchSize for model

雾聪

2024-03-29 3ac539a58ca56a04123771ef145430cbcbdc294e

 funasr/models/conformer/encoder.py

@@ -47,7 +47,7 @@
from funasr.models.transformer.utils.subsampling import Conv2dSubsamplingPad
from funasr.models.transformer.utils.subsampling import StreamingConvInput
from funasr.register import tables

import pdb

class ConvolutionModule(nn.Module):
    """ConvolutionModule in Conformer model.