From 980007a486a4a2cfbcf4f98c7a88e28e17fd0f48 Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期四, 21 三月 2024 14:59:46 +0800
Subject: [PATCH] update seaco finetune (#1526)
---
funasr/train_utils/average_nbest_models.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/train_utils/average_nbest_models.py b/funasr/train_utils/average_nbest_models.py
index f117804..3603a44 100644
--- a/funasr/train_utils/average_nbest_models.py
+++ b/funasr/train_utils/average_nbest_models.py
@@ -143,7 +143,7 @@
return checkpoint_paths
@torch.no_grad()
-def average_checkpoints(output_dir: str, last_n: int=5):
+def average_checkpoints(output_dir: str, last_n: int=5, val_acc_list=[]):
"""
Average the last 'last_n' checkpoints' model state_dicts.
If a tensor is of type torch.int, perform sum instead of average.
--
Gitblit v1.9.1