mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
update
This commit is contained in:
parent
622d799de8
commit
3e8b21a9f1
@ -4,9 +4,6 @@ import torch
|
|||||||
from torch.cuda.amp import autocast
|
from torch.cuda.amp import autocast
|
||||||
from typing import Union, Dict, List, Tuple, Optional
|
from typing import Union, Dict, List, Tuple, Optional
|
||||||
|
|
||||||
from funasr.register import tables
|
|
||||||
from funasr.utils import postprocess_utils
|
|
||||||
from funasr.utils.datadir_writer import DatadirWriter
|
|
||||||
from funasr.models.paraformer.cif_predictor import mae_loss
|
from funasr.models.paraformer.cif_predictor import mae_loss
|
||||||
from funasr.train_utils.device_funcs import force_gatherable
|
from funasr.train_utils.device_funcs import force_gatherable
|
||||||
from funasr.models.transformer.utils.add_sos_eos import add_sos_eos
|
from funasr.models.transformer.utils.add_sos_eos import add_sos_eos
|
||||||
@ -19,7 +16,6 @@ from funasr.models.ctc.ctc import CTC
|
|||||||
from funasr.utils.load_utils import load_audio_and_text_image_video, extract_fbank
|
from funasr.utils.load_utils import load_audio_and_text_image_video, extract_fbank
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@tables.register("model_classes", "monotonicaligner")
|
@tables.register("model_classes", "monotonicaligner")
|
||||||
class MonotonicAligner(torch.nn.Module):
|
class MonotonicAligner(torch.nn.Module):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user