Commit be938e7e authored by Zhaoheng Ni's avatar Zhaoheng Ni Committed by Facebook GitHub Bot
Browse files

Fix HuBERT docstring (#2746)

Summary:
The docstring of `wav2vec2` argument is wrong. Fix it in this PR.

Pull Request resolved: https://github.com/pytorch/audio/pull/2746

Reviewed By: carolineechen

Differential Revision: D40225995

Pulled By: nateanl

fbshipit-source-id: 770e9c928ebebd7b6307e181601eb64625d668da
parent 0b4b1fd4
...@@ -137,12 +137,8 @@ class HuBERTPretrainModel(Module): ...@@ -137,12 +137,8 @@ class HuBERTPretrainModel(Module):
<https://github.com/pytorch/audio/tree/release/0.12/examples/hubert>`__ <https://github.com/pytorch/audio/tree/release/0.12/examples/hubert>`__
Args: Args:
feature_extractor (torch.nn.Module): wav2vec2 (Wav2Vec2Model):
Feature extractor that extracts feature vectors from raw audio Tensor. Wav2Vec2 encoder that generates the transformer outputs.
encoder (torch.nn.Module):
Encoder that converts the audio features into the sequence of probability
distribution (in negative log-likelihood) over labels.
mask_generator (torch.nn.Module): mask_generator (torch.nn.Module):
Mask generator that generates the mask for masked prediction during the training. Mask generator that generates the mask for masked prediction during the training.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment