start_data_split_i=0

2025-09-15 14:48:36 +08:00 · 2024-05-09 10:15:44 +08:00 · 2024-05-09 10:15:44 +08:00 · f775e0e41c
commit f775e0e41c
parent 27062fecc1
1 changed files with 1 additions and 0 deletions
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@ -223,6 +223,7 @@ def main(**kwargs):

            torch.cuda.empty_cache()

+        trainer.start_data_split_i = 0
        trainer.validate_epoch(
            model=model, dataloader_val=dataloader_val, epoch=epoch + 1, writer=writer
        )