"tests/vscode:/vscode.git/clone" did not exist on "b09912c8f452ac485933ac0f86937aa01de3c398"
Unverified Commit 05b8e25f authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

[VideoMAE] Fix docstring (#21111)



Fix docstring
Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
parent 4ed89d48
......@@ -898,8 +898,8 @@ class VideoMAEForPreTraining(VideoMAEPreTrainedModel):
@add_start_docstrings(
"""VideoMAE Model transformer with a video classification head on top (a linear layer on top of the final hidden state of
the [CLS] token) e.g. for ImageNet.""",
"""VideoMAE Model transformer with a video classification head on top (a linear layer on top of the average pooled hidden
states of all tokens) e.g. for ImageNet.""",
VIDEOMAE_START_DOCSTRING,
)
class VideoMAEForVideoClassification(VideoMAEPreTrainedModel):
......
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