From ace42e5043be992c3a324fc637ba0594f10ad3b6 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期五, 08 十二月 2023 16:39:38 +0800
Subject: [PATCH] fix loss normalization for ddp training
---
funasr/bin/lm_inference_launch.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/lm_inference_launch.py b/funasr/bin/lm_inference_launch.py
index 236a923..f12f50a 100644
--- a/funasr/bin/lm_inference_launch.py
+++ b/funasr/bin/lm_inference_launch.py
@@ -183,6 +183,7 @@
dtype=dtype,
batch_size=batch_size,
key_file=key_file,
+ preprocess_fn=preprocessor,
num_workers=num_workers,
)
--
Gitblit v1.9.1