File size: 454 Bytes
c540376 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
---
license: apache-2.0
---
Model: ViTAE-RVSA (https://arxiv.org/abs/2208.03987)
Variant: ViTAE-b_pretrain
Example Usage:
```python
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)
``` |