From 3f487c42904a27deeae4ab48cf8ccc45537263d1 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 31 一月 2024 16:21:10 +0800
Subject: [PATCH] funasr1.0.5 (#1328)
---
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