add SetBatchSize for model

雾聪

2024-03-29 3ac539a58ca56a04123771ef145430cbcbdc294e

 funasr/models/sond/encoder/conv_encoder.py

@@ -12,7 +12,7 @@
from funasr.models.transformer.layer_norm import LayerNorm
from funasr.models.encoder.abs_encoder import AbsEncoder
import math
from funasr.models.transformer.repeat import repeat
from funasr.models.transformer.utils.repeat import repeat


class EncoderLayer(nn.Module):