speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update train recipe

Browse Source
This commit is contained in:
游雁 2024-02-21 14:34:36 +08:00
parent 26ab38b56c
commit 7357bb6c45
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/train_utils/trainer.py
View File

@ -157,7 +157,7 @@ class Trainer:
self._resume_checkpoint(self.output_dir)
for epoch in range(self.start_epoch, self.max_epoch + 1):
time1 = time.perf_counter()
self._train_epoch(epoch)
@ -179,6 +179,9 @@ class Trainer:
self.scheduler.step()
time2 = time.perf_counter()
time_escaped = (time2 - time1)/3600.0
print(f"time_escaped_epoch: {time_escaped:.3f} hours, estimated to finish: {(self.max_epoch-epoch)*time_escaped:.3f}")
if self.rank == 0:
average_checkpoints(self.output_dir, self.avg_nbest_model)
@ -285,7 +288,7 @@ class Trainer:
)
lr = self.scheduler.get_last_lr()[0]
time_now = datetime.now()
time_now = now.strftime("%Y-%m-%d %H:%M:%S")
time_now = time_now.strftime("%Y-%m-%d %H:%M:%S")
description = (
f"{time_now}, "
f"rank: {self.local_rank}, "