From 8ea5f1302d1eb64ac1c00799c83b50f15c793633 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Wed, 21 Feb 2024 14:55:30 +0800 Subject: [PATCH] update train recipe --- funasr/train_utils/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py index d24023d21..c2326424f 100644 --- a/funasr/train_utils/trainer.py +++ b/funasr/train_utils/trainer.py @@ -108,7 +108,7 @@ class Trainer: filename = os.path.join(self.output_dir, f'model.pt.ep{epoch}') 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')) torch.save(state, latest) @@ -181,7 +181,7 @@ class Trainer: 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}") + 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: average_checkpoints(self.output_dir, self.avg_nbest_model)