Skip to content

GitLab

Unverified Commit 4130e703 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

VideoMAE doctest - use valid dummy pixel values (#22022) 

Use valid dummy pixel values
parent c1f85598
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/videomae/modeling_videomae.py
View file @ 4130e703
...@@ -790,7 +790,7 @@ class VideoMAEForPreTraining(VideoMAEPreTrainedModel): ...@@ -790,7 +790,7 @@ class VideoMAEForPreTraining(VideoMAEPreTrainedModel):
>>> import torch >>> import torch
>>> num_frames = 16 >>> num_frames = 16
>>> video = list(np.random.randn(16, 3, 224, 224)) >>> video = list(np.random.randint(0, 256, (num_frames, 3, 224, 224)))
>>> image_processor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base") >>> image_processor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base")
>>> model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base") >>> model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
......
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