Srikumar26 commited on
Commit
fc302c5
1 Parent(s): d6bd18e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+ Model: SatMAE (https://arxiv.org/abs/2207.08051)
5
+
6
+ Variant: vitlarge-fmow-pretrain-800
7
+
8
+ Example Usage:
9
+ ```python
10
+ from huggingface_hub import hf_hub_download
11
+ import torch
12
+
13
+ hf_hub_download("MVRL/satmae-vitlarge-fmow-pretrain-800", "model.py", local_dir=".")
14
+
15
+ from model import MaskedAutoencoderViT
16
+
17
+ model = MaskedAutoencoderViT.from_pretrained("MVRL/satmae-vitlarge-fmow-pretrain-800")
18
+
19
+ print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shape)
20
+ ```