riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

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

Browse Source 
Use valid dummy pixel values
This commit is contained in:
amyeroberts 2023-03-08 11:54:42 +00:00 committed by GitHub
parent c1f85598eb
commit 4130e70367
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/videomae/modeling_videomae.py
View File

@ -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")