From cc10d01fd7d2ac51a890550d6dd67109cf7cd069 Mon Sep 17 00:00:00 2001 From: 雾聪 <wucong.lyb@alibaba-inc.com> Date: 星期五, 29 三月 2024 17:01:28 +0800 Subject: [PATCH] fix func SetBatchSize --- funasr/models/eres2net/eres2net_aug.py | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/funasr/models/eres2net/eres2net_aug.py b/funasr/models/eres2net/eres2net_aug.py index 23e6b8f..8e4fc21 100644 --- a/funasr/models/eres2net/eres2net_aug.py +++ b/funasr/models/eres2net/eres2net_aug.py @@ -16,6 +16,7 @@ import torch.nn.functional as F import funasr.models.sond.pooling.pooling_layers as pooling_layers + from funasr.models.eres2net.fusion import AFF -- Gitblit v1.9.1