From ee9e86d5f0b5f546f77de1ea8b59cdb67aaad478 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 25 四月 2023 11:32:41 +0800
Subject: [PATCH] update

---
 funasr/datasets/small_datasets/preprocessor.py |    2 +-
 funasr/bin/train.py                            |    1 +
 2 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index b8cbf64..4641370 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -527,6 +527,7 @@
                                                                    distributed_option.dist_rank,
                                                                    distributed_option.local_rank))
     logging.info(pytorch_cudnn_version())
+    logging.info("Args: {}".format(args))
     logging.info(model_summary(model))
     logging.info("Optimizer: {}".format(optimizers))
     logging.info("Scheduler: {}".format(schedulers))
diff --git a/funasr/datasets/small_datasets/preprocessor.py b/funasr/datasets/small_datasets/preprocessor.py
index ecd4478..7ec6929 100644
--- a/funasr/datasets/small_datasets/preprocessor.py
+++ b/funasr/datasets/small_datasets/preprocessor.py
@@ -820,7 +820,7 @@
 
 
 def build_preprocess(args, train):
-    if args.use_preprocessor:
+    if not args.use_preprocessor:
         return None
     if args.task_name in ["asr", "data2vec", "diar", "sv"]:
         retval = CommonPreprocessor(

--
Gitblit v1.9.1