From db77a41e29d5c190d1bbc0d7698e384f14fa8168 Mon Sep 17 00:00:00 2001
From: yhliang <429259365@qq.com>
Date: 星期四, 18 五月 2023 11:11:19 +0800
Subject: [PATCH] fix m2met2 baseline bugs

---
 funasr/models/e2e_sa_asr.py |    3 +--
 1 files changed, 1 insertions(+), 2 deletions(-)

diff --git a/funasr/models/e2e_sa_asr.py b/funasr/models/e2e_sa_asr.py
index 0d4097e..f694cc2 100644
--- a/funasr/models/e2e_sa_asr.py
+++ b/funasr/models/e2e_sa_asr.py
@@ -16,9 +16,8 @@
 
 from funasr.layers.abs_normalize import AbsNormalize
 from funasr.losses.label_smoothing_loss import (
-    LabelSmoothingLoss,  # noqa: H301
+    LabelSmoothingLoss, NllLoss  # noqa: H301
 )
-from funasr.losses.nll_loss import NllLoss
 from funasr.models.ctc import CTC
 from funasr.models.decoder.abs_decoder import AbsDecoder
 from funasr.models.encoder.abs_encoder import AbsEncoder

--
Gitblit v1.9.1