funasr/bin/train.py
@@ -223,6 +223,7 @@ torch.cuda.empty_cache() trainer.start_data_split_i = 0 trainer.validate_epoch( model=model, dataloader_val=dataloader_val, epoch=epoch + 1, writer=writer ) @@ -240,6 +241,8 @@ f"estimated to finish {trainer.max_epoch} " f"epoch: {(trainer.max_epoch - epoch) * time_escaped:.3f} hours\n" ) trainer.train_acc_avg = 0.0 trainer.train_loss_avg = 0.0 if trainer.rank == 0: average_checkpoints(trainer.output_dir, trainer.avg_nbest_model)
