From 4152cf4615e234b9892703a8e088cdd51937fc13 Mon Sep 17 00:00:00 2001 From: haoneng.lhn <haoneng.lhn@alibaba-inc.com> Date: 星期四, 07 十二月 2023 14:49:34 +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