TaoRuijie / ECAPA-TDNN

Unofficial reimplementation of ECAPA-TDNN for speaker recognition (EER=0.86 for Vox1_O when train only in Vox2)
MIT License
581 stars 111 forks source link

save model #54

Closed phchtl closed 1 year ago

phchtl commented 1 year ago

i want to ask can we save the model as .pth file,and how should i do? Thank you so much

TaoRuijie commented 1 year ago

I think you can check the function of torch.load and torch.save, I think just change the name is fine ?

phchtl commented 1 year ago

Thank you so much