From b81396f912456a6a9d3813467242a805e7ce7553 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 01 四月 2024 14:31:48 +0800
Subject: [PATCH] add dec reset for batch
---
funasr/bin/train.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 569757a..3c93371 100644
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -35,7 +35,7 @@
assert "model" in kwargs
if "model_conf" not in kwargs:
- logging.info("download models from model hub: {}".format(kwargs.get("model_hub", "ms")))
+ logging.info("download models from model hub: {}".format(kwargs.get("hub", "ms")))
kwargs = download_model(is_training=kwargs.get("is_training", True), **kwargs)
--
Gitblit v1.9.1