hitachi-speech / EEND

End-to-End Neural Diarization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Question about pit loss calculation in EDA-EEND model

SoundingSilence opened this issue · comments

commented

Thanks for your valuable work on EEND. In eend/chainer_backend/models.py (line 491-497), why the loss is calculated twice, substituted with standard loss finally?

loss, labels = batch_pit_n_speaker_loss(ys_padded, ts_padded, n_speakers)

loss = standard_loss(ys, labels)

I am looking forward to your reply. Thanks!