From bd7455ec7da3178d9acc8d704ee63cb443a8887e Mon Sep 17 00:00:00 2001 From: speech_asr <wangjiaming.wjm@alibaba-inc.com> Date: 星期三, 12 四月 2023 10:43:01 +0800 Subject: [PATCH] update --- funasr/tasks/abs_task.py | 6 +++--- 1 files changed, 3 insertions(+), 3 deletions(-) diff --git a/funasr/tasks/abs_task.py b/funasr/tasks/abs_task.py index 775cba8..86957d9 100644 --- a/funasr/tasks/abs_task.py +++ b/funasr/tasks/abs_task.py @@ -30,6 +30,7 @@ import torch.nn import torch.optim import yaml +from funasr.train.abs_espnet_model import AbsESPnetModel from torch.utils.data import DataLoader from typeguard import check_argument_types from typeguard import check_return_type @@ -44,19 +45,18 @@ from funasr.iterators.multiple_iter_factory import MultipleIterFactory from funasr.iterators.sequence_iter_factory import SequenceIterFactory from funasr.main_funcs.collect_stats import collect_stats -from funasr.optimizers.sgd import SGD from funasr.optimizers.fairseq_adam import FairseqAdam +from funasr.optimizers.sgd import SGD from funasr.samplers.build_batch_sampler import BATCH_TYPES from funasr.samplers.build_batch_sampler import build_batch_sampler from funasr.samplers.unsorted_batch_sampler import UnsortedBatchSampler from funasr.schedulers.noam_lr import NoamLR -from funasr.schedulers.warmup_lr import WarmupLR from funasr.schedulers.tri_stage_scheduler import TriStageLR +from funasr.schedulers.warmup_lr import WarmupLR from funasr.torch_utils.load_pretrained_model import load_pretrained_model from funasr.torch_utils.model_summary import model_summary from funasr.torch_utils.pytorch_version import pytorch_cudnn_version from funasr.torch_utils.set_all_random_seed import set_all_random_seed -from funasr.train.abs_espnet_model import AbsESPnetModel from funasr.train.class_choices import ClassChoices from funasr.train.distributed_utils import DistributedOption from funasr.train.trainer import Trainer -- Gitblit v1.9.1