MVRL
/

scalemae-vitlarge-800

Model card Files Files and versions

Srikumar26 commited on May 23, 2024

Commit

e323e80

·

verified ·

1 Parent(s): ff94ddd

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ hf_hub_download("MVRL/scalemae-vitlarge-800", "model.py", local_dir=".")
 from model import ScaleMAE_baseline
-model = MaskedAutoencoderViT.from_pretrained("MVRL/scalemae-vitlarge-800")
 print(model(torch.randn(1, 3, 224, 224),patch_size=16).shape)
 ```

 from model import ScaleMAE_baseline
+model = ScaleMAE_baseline.from_pretrained("MVRL/scalemae-vitlarge-800")
 print(model(torch.randn(1, 3, 224, 224),patch_size=16).shape)
 ```