From d8ef0e9eb438ac8e5264b7868e29b47588f4be3b Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 22 一月 2024 21:14:30 +0800
Subject: [PATCH] Funasr1.0 (#1282)
---
funasr/models/bat/cif_predictor.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models/bat/cif_predictor.py b/funasr/models/bat/cif_predictor.py
index 9aa3e33..d8915c2 100644
--- a/funasr/models/bat/cif_predictor.py
+++ b/funasr/models/bat/cif_predictor.py
@@ -8,7 +8,7 @@
# from funasr.models.scama.utils import sequence_mask
# from typing import Optional, Tuple
#
-# from funasr.utils.register import register_class
+# from funasr.register import tables
#
# class mae_loss(nn.Module):
#
@@ -93,7 +93,7 @@
# fires = torch.stack(list_fires, 1)
# return fires
#
-# @register_class("predictor_classes", "BATPredictor")
+# @tables.register("predictor_classes", "BATPredictor")
# class BATPredictor(nn.Module):
# def __init__(self, idim, l_order, r_order, threshold=1.0, dropout=0.1, smooth_factor=1.0, noise_threshold=0, return_accum=False):
# super(BATPredictor, self).__init__()
--
Gitblit v1.9.1