zycv / OpenSpeaker

OpenSpeaker is a completely independent and open source speaker recognition project. It provides the entire process of speaker recognition including multi-platform deployment and model optimization.
Apache License 2.0
61 stars 12 forks source link

Is there a way to export sb model to jit model? #1

Closed fanlu closed 3 years ago

zycv commented 3 years ago

Yes, In this repo I have uploaded a scripts to convert ecapa or tdnn in speechbrain to jit model,which can be found here: https://github.com/zycv/speechbrain/blob/OpenSpeaker/recipes/VoxCeleb/SpeakerRec/tools/convert_to_torchscript.py