From 16a976a01d110d3969759be7720cae2b6b0664f7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期日, 24 三月 2024 01:27:08 +0800
Subject: [PATCH] finetune
---
funasr/bin/train.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 0ff4ba1..5cf54da 100644
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -128,7 +128,8 @@
else:
model = model.to(device=kwargs.get("device", "cuda"))
- logging.info(f"{model}")
+ if local_rank == 0:
+ logging.info(f"{model}")
kwargs["device"] = next(model.parameters()).device
# optim
--
Gitblit v1.9.1