From f5958e96da05220c9e07d0af63df339cae4cc6f7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 30 四月 2024 23:09:26 +0800
Subject: [PATCH] train_loss_avg train_acc_avg
---
funasr/datasets/sense_voice_datasets/datasets.py | 2 +-
funasr/train_utils/trainer.py | 1 +
2 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/datasets/sense_voice_datasets/datasets.py b/funasr/datasets/sense_voice_datasets/datasets.py
index 1d269dd..dfadbab 100644
--- a/funasr/datasets/sense_voice_datasets/datasets.py
+++ b/funasr/datasets/sense_voice_datasets/datasets.py
@@ -186,7 +186,7 @@
)
if self.batch_type != "example":
- for i in range(3):
+ for i in range(10):
outputs = self._filter_badcase(outputs, i=i)
return outputs
diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py
index dd0ac7a..e0b6def 100644
--- a/funasr/train_utils/trainer.py
+++ b/funasr/train_utils/trainer.py
@@ -308,6 +308,7 @@
checkpoint["step_in_epoch"] if "step_in_epoch" in checkpoint else 0
)
self.step_in_epoch = 0 if self.step_in_epoch is None else self.step_in_epoch
+ print(checkpoint["train_acc_avg"])
self.train_acc_avg = (
checkpoint["train_acc_avg"] if "train_acc_avg" in checkpoint else 0
)
--
Gitblit v1.9.1