From 8fe31a144a89abe581f3f1f64a39c2e934eb5f29 Mon Sep 17 00:00:00 2001
From: Yabin Li <wucong.lyb@alibaba-inc.com>
Date: 星期三, 20 三月 2024 19:12:16 +0800
Subject: [PATCH] Update SDK_advanced_guide_online_zh.md
---
funasr/train_utils/trainer.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py
index a00b3de..14abd6c 100644
--- a/funasr/train_utils/trainer.py
+++ b/funasr/train_utils/trainer.py
@@ -146,7 +146,7 @@
"""
ckpt = os.path.join(resume_path, "model.pt")
if os.path.isfile(ckpt):
- checkpoint = torch.load(ckpt)
+ checkpoint = torch.load(ckpt, map_location="cpu")
self.start_epoch = checkpoint['epoch'] + 1
# self.model.load_state_dict(checkpoint['state_dict'])
src_state = checkpoint['state_dict']
@@ -169,7 +169,8 @@
print(f"Checkpoint loaded successfully from '{ckpt}'")
else:
print(f"No checkpoint found at '{ckpt}', does not resume status!")
-
+
+ self.model.to(self.device)
if self.use_ddp or self.use_fsdp:
dist.barrier()
--
Gitblit v1.9.1