From 4800870825aa2ee4430280b6a6a3f6cabf2d68b7 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Wed, 3 Aug 2022 10:17:01 +0000 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 217c548..1cdbeee 100644 --- a/README.md +++ b/README.md @@ -34,8 +34,8 @@ import torch video = list(np.random.randn(16, 3, 224, 224)) -feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base") -model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base") +feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") +model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") inputs = feature_extractor(video, return_tensors="pt")