From acb9a0fec8d8a4dabeedcbb8e08c26f66d7083f0 Mon Sep 17 00:00:00 2001 From: haoneng.lhn <haoneng.lhn@alibaba-inc.com> Date: 星期五, 08 十二月 2023 16:19:00 +0800 Subject: [PATCH] fix loss normalization for ddp training --- funasr/bin/tp_infer.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/bin/tp_infer.py b/funasr/bin/tp_infer.py index ede579c..cfe534f 100644 --- a/funasr/bin/tp_infer.py +++ b/funasr/bin/tp_infer.py @@ -11,7 +11,7 @@ import torch from funasr.build_utils.build_model_from_file import build_model_from_file from funasr.models.frontend.wav_frontend import WavFrontend -from funasr.text.token_id_converter import TokenIDConverter +from funasr.tokenizer.token_id_converter import TokenIDConverter from funasr.torch_utils.device_funcs import to_device -- Gitblit v1.9.1