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): ...@@ -898,8 +898,8 @@ class VideoMAEForPreTraining(VideoMAEPreTrainedModel):
@add_start_docstrings( @add_start_docstrings(
"""VideoMAE Model transformer with a video classification head on top (a linear layer on top of the final hidden state of """VideoMAE Model transformer with a video classification head on top (a linear layer on top of the average pooled hidden
the [CLS] token) e.g. for ImageNet.""", states of all tokens) e.g. for ImageNet.""",
VIDEOMAE_START_DOCSTRING, VIDEOMAE_START_DOCSTRING,
) )
class VideoMAEForVideoClassification(VideoMAEPreTrainedModel): 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