R1ckShi
2024-06-11 a0c58c8fc6fd68055897be95d11be35506581a91
funasr/models/normalize/utterance_mvn.py
@@ -3,9 +3,10 @@
import torch
from funasr.models.transformer.utils.nets_utils import make_pad_mask
from funasr.utils.register import register_class, registry_tables
from funasr.register import tables
@register_class("normalize_classes", "UtteranceMVN")
@tables.register("normalize_classes", "UtteranceMVN")
class UtteranceMVN(torch.nn.Module):
    def __init__(
        self,