当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch import_huggingface_model用法及代码示例


本文简要介绍python语言中 torchaudio.models.wav2vec2.utils.import_huggingface_model 的用法。

用法:

torchaudio.models.wav2vec2.utils.import_huggingface_model(original: torch.nn.Module) → torchaudio.models.Wav2Vec2Model

参数

original(torch.nn.Module) -来自 transformersWav2Vec2ForCTC 的实例。

返回

导入型号。

返回类型

Wav2Vec2模型

从 Hugging Face 的 Transformers 的相应模型对象构建 Wav2Vec2Model 。

示例
>>> from torchaudio.models.wav2vec2.utils import import_huggingface_model
>>>
>>> original = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
>>> model = import_huggingface_model(original)
>>>
>>> waveforms, _ = torchaudio.load("audio.wav")
>>> logits, _ = model(waveforms)

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torchaudio.models.wav2vec2.utils.import_huggingface_model。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。