mindspore-lab / mindaudio

A toolbox of audio models and algorithms based on MindSpore

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

预训练的模型怎么加载使用和提取向量呢

lonngxiang opened this issue · comments

1.下载weights下的ckpt文件,在yaml配置文件中将pre_trained设置为True,并配置正确的checkpoint_path即可加载预训练模型。

2.from mindaudio.models.ecapatdnn import EcapaTDNN,并如speaker_verification_cosine.py文件中加载训练好的模型,使用此函数进行推理,即可提取向量。