From 5602ebe208639ad8f91899adeddae0a2f1e39f09 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期三, 31 一月 2024 10:07:34 +0800
Subject: [PATCH] code update
---
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