From 45dc927a97831c85a422fee5ee3010b2dc0ce3c7 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期日, 04 二月 2024 13:52:00 +0800
Subject: [PATCH] update docs
---
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