From c892cc34a9e181e9ea7b4e59c35651a61149401f Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 23 一月 2024 13:48:06 +0800
Subject: [PATCH] funasr1.0.1
---
funasr/train_utils/trainer.py | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py
index 62d6be8..414c0d7 100644
--- a/funasr/train_utils/trainer.py
+++ b/funasr/train_utils/trainer.py
@@ -148,6 +148,7 @@
self._train_epoch(epoch)
+
if self.use_ddp or self.use_fsdp:
dist.barrier()
@@ -156,8 +157,8 @@
if self.use_ddp or self.use_fsdp:
dist.barrier()
-
-
+
+
if self.rank == 0:
self._save_checkpoint(epoch)
@@ -172,7 +173,8 @@
if self.use_ddp or self.use_fsdp:
dist.barrier()
-
+
+
if self.writer:
self.writer.close()
--
Gitblit v1.9.1