funasr/bin/train.py
@@ -198,7 +198,7 @@ writer = None dataloader_tr, dataloader_val = None, None for epoch in range(trainer.start_epoch, trainer.max_epoch + 1): for epoch in range(trainer.start_epoch, trainer.max_epoch): time1 = time.perf_counter() for data_split_i in range(trainer.start_data_split_i, dataloader.data_split_num):