From 2cf4084b23db9bd9e8ce4db76d0628ef6655ed71 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期六, 03 二月 2024 13:05:07 +0800
Subject: [PATCH] fix retract error (#1350)
---
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