From 930fe72f43524b4e355ef671c7180cc6cf9eefb5 Mon Sep 17 00:00:00 2001 From: 雾聪 <wucong.lyb@alibaba-inc.com> Date: 星期一, 01 四月 2024 15:27:20 +0800 Subject: [PATCH] set batch default value --- funasr/models/sond/encoder/conv_encoder.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/models/sond/encoder/conv_encoder.py b/funasr/models/sond/encoder/conv_encoder.py index 4c345cb..3933c01 100644 --- a/funasr/models/sond/encoder/conv_encoder.py +++ b/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): -- Gitblit v1.9.1