search

Audio-WestlakeU / FS-EEND

The official Pytorch implementation of "Frame-wise streaming end-to-end speaker diarization with non-autoregressive self-attention-based attractors". [ICASSP 2024]
MIT License
71 stars 4 forks source link

inference #3

Closed DAYTOY-1112 closed 8 months ago

DAYTOY-1112 commented 9 months ago

老师您好，推理时运行python train_diaxxx.py文件是python train_dia.py这个代码吗

DiLiangWU commented 9 months ago

train_dia.py对应simu dataset的训练和推理，train_dia_fintn_ch.py对应在callhome数据集上的finetune和推理。如果您希望应用在其他数据集上，请先fintune。