预训练的模型怎么加载使用和提取向量呢
lonngxiang opened this issue · comments
dragon10 commented
litingyu commented
1.下载weights下的ckpt文件,在yaml配置文件中将pre_trained设置为True,并配置正确的checkpoint_path即可加载预训练模型。
litingyu commented
2.from mindaudio.models.ecapatdnn import EcapaTDNN,并如speaker_verification_cosine.py文件中加载训练好的模型,使用此函数进行推理,即可提取向量。