Views
No views yet
1from huggingface_hub import hf_hub_download
2import torch
3
4hf_hub_download("MVRL/scalemae-vitlarge-800", "model.py", local_dir=".")
5
6from model import ScaleMAE_baseline
7
8model = ScaleMAE_baseline.from_pretrained("MVRL/scalemae-vitlarge-800")
9
10print(model(torch.randn(1, 3, 224, 224),patch_size=16).shape)