zhifu gao
2024-04-08 6ddd383370c3b264c5c56a8fafb33e5cde20cfc5
funasr/bin/train.py
@@ -214,7 +214,7 @@
    if trainer.rank == 0:
        average_checkpoints(trainer.output_dir, trainer.avg_nbest_model, trainer.val_acc_list)
        average_checkpoints(trainer.output_dir, trainer.avg_nbest_model)
    trainer.close()