Downloads · 30 days
6
2% of all-time downloads
MVRL/rvsa_vitae_b
rvsa_vitae_b is a machine learning model from MVRL. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
Model: ViTAE-RVSA (https://arxiv.org/abs/2208.03987)
Downloads · 30 days
6
2% of all-time downloads
All-time downloads
291
Public
Repo size
918 MB
Likes
0
Public
Click a slice to open those files.
.bin459 MB · 100%
From the Hugging Face model README
Model: ViTAE-RVSA (https://arxiv.org/abs/2208.03987)
Variant: ViTAE-b_pretrain
Example Usage:
from huggingface_hub import hf_hub_download
import torch
hf_hub_download("MVRL/rvsa_vitae_b", "model.py", local_dir=".")
from model import MaskedAutoencoderViTAE
model = MaskedAutoencoderViTAE.from_pretrained("MVRL/rvsa_vitae_b")
print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shape)