From f5f8f81ce9ff288b499b474ee1181cab41426cda Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 29 三月 2024 09:59:34 +0800
Subject: [PATCH] Dev gzf new (#1561)
---
funasr/train_utils/average_nbest_models.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/train_utils/average_nbest_models.py b/funasr/train_utils/average_nbest_models.py
index 013a719..c31c1cb 100644
--- a/funasr/train_utils/average_nbest_models.py
+++ b/funasr/train_utils/average_nbest_models.py
@@ -23,10 +23,10 @@
in the output directory.
"""
try:
- checkpoint = torch.load(os.path.exists(os.path.join(output_dir, "model.pt")), map_location="cpu")
+ checkpoint = torch.load(os.path.join(output_dir, "model.pt"), map_location="cpu")
avg_keep_nbest_models_type = checkpoint["avg_keep_nbest_models_type"]
val_step_or_eoch = checkpoint[f"val_{avg_keep_nbest_models_type}_step_or_eoch"]
- sorted_items = sorted(saved_ckpts.items(), key=lambda x: x[1], reverse=True)
+ sorted_items = sorted(val_step_or_eoch.items(), key=lambda x: x[1], reverse=True)
sorted_items = sorted_items[:last_n] if avg_keep_nbest_models_type == "acc" else sorted_items[-last_n:]
checkpoint_paths = [os.path.join(output_dir, key) for key, value in sorted_items[:last_n]]
except:
--
Gitblit v1.9.1