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) -來自 transformers 的 Wav2Vec2ForCTC 的實例。

返回：

導入型號。

返回類型：

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)