facebook
/
timesformer-base-finetuned-k400
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Niels Rogge 2023-01-02 11:43:07 +00:00 committed by huggingface-web
parent e040485005
commit 8aaf40ea7d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -26,10 +26,10 @@ import torch
video = list(np.random.randn(8, 3, 224, 224)) video = list(np.random.randn(8, 3, 224, 224))
feature_extractor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400") processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400") model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
inputs = feature_extractor(video, return_tensors="pt") inputs = processor(video, return_tensors="pt")
with torch.no_grad(): with torch.no_grad():
outputs = model(**inputs) outputs = model(**inputs)