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:55:30 +08:00
parent ed35838e89
commit 8ea5f1302d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/train_utils/trainer.py
View File

@ -108,7 +108,7 @@ class Trainer:
filename = os.path.join(self.output_dir, f'model.pt.ep{epoch}') filename = os.path.join(self.output_dir, f'model.pt.ep{epoch}')
torch.save(state, filename) torch.save(state, filename)
print(f'Checkpoint saved to {filename}') print(f'\nCheckpoint saved to {filename}\n')
latest = Path(os.path.join(self.output_dir, f'model.pt')) latest = Path(os.path.join(self.output_dir, f'model.pt'))
torch.save(state, latest) torch.save(state, latest)
@ -181,7 +181,7 @@ class Trainer:
time2 = time.perf_counter() time2 = time.perf_counter()
time_escaped = (time2 - time1)/3600.0 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}") print(f"\ntime_escaped_epoch: {time_escaped:.3f} hours, estimated to finish {self.max_epoch} epoch: {(self.max_epoch-epoch)*time_escaped:.3f}\n")
if self.rank == 0: if self.rank == 0:
average_checkpoints(self.output_dir, self.avg_nbest_model) average_checkpoints(self.output_dir, self.avg_nbest_model)