From c62d0537a6757754631970259f15cbf4d8c01765 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Thu, 8 Aug 2024 17:29:09 +0800 Subject: [PATCH] add --- funasr/train_utils/trainer_ds.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py index 9ef9dc9e4..ea3fb685b 100644 --- a/funasr/train_utils/trainer_ds.py +++ b/funasr/train_utils/trainer_ds.py @@ -895,6 +895,13 @@ class Trainer: if writer is not None: writer.add_scalar(f"rank{self.rank}_loss/{tag}", loss, batch_total) writer.add_scalar(f"rank{self.rank}_lr/{tag}", lr, batch_total) + writer.add_scalar( + f"rank{self.rank}_acc_avg_slice/{tag}", acc_avg_epoch, batch_total + ) + writer.add_scalar( + f"rank{self.rank}_loss_avg_epoch/{tag}", loss_avg_epoch, batch_total + ) + for key, var in stats.items(): writer.add_scalar(f"stats_rank{self.rank}_{key}/{tag}", var.item(), batch_total) description_dict[f"stats_rank{self.rank}_{key}/{tag}"] = var.item()