From 08e6f946aa4c02e3ba3cebb47af2ac3cd5abe97b Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 05 二月 2024 11:09:59 +0800
Subject: [PATCH] disable_log
---
funasr/bin/train.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 0334006..8ea0c0d 100644
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -40,7 +40,7 @@
def main(**kwargs):
-
+ print(kwargs)
# set random seed
tables.print()
set_all_random_seed(kwargs.get("seed", 0))
@@ -154,7 +154,7 @@
if batch_sampler is not None:
batch_sampler_class = tables.batch_sampler_classes.get(batch_sampler)
batch_sampler = batch_sampler_class(dataset_tr, **kwargs.get("dataset_conf"))
- batch_sampler_val = batch_sampler_class(dataset_tr, is_training=False, **kwargs.get("dataset_conf"))
+ batch_sampler_val = batch_sampler_class(dataset_val, is_training=False, **kwargs.get("dataset_conf"))
dataloader_tr = torch.utils.data.DataLoader(dataset_tr,
collate_fn=dataset_tr.collator,
batch_sampler=batch_sampler,
--
Gitblit v1.9.1