Commit
•
a9ba7f4
1
Parent(s):
fc302c5
Push model using huggingface_hub.
Browse files- README.md +6 -17
- config.json +13 -0
- model.safetensors +3 -0
README.md
CHANGED
|
@@ -1,20 +1,9 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
-
Model: SatMAE (https://arxiv.org/abs/2207.08051)
|
| 5 |
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 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 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
+
tags:
|
| 3 |
+
- pytorch_model_hub_mixin
|
| 4 |
+
- model_hub_mixin
|
| 5 |
---
|
|
|
|
| 6 |
|
| 7 |
+
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|
| 8 |
+
- Library: [More Information Needed]
|
| 9 |
+
- Docs: [More Information Needed]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"decoder_depth": 8,
|
| 3 |
+
"decoder_embed_dim": 512,
|
| 4 |
+
"decoder_num_heads": 16,
|
| 5 |
+
"depth": 24,
|
| 6 |
+
"embed_dim": 1024,
|
| 7 |
+
"img_size": 224,
|
| 8 |
+
"in_c": 3,
|
| 9 |
+
"mlp_ratio": 4.0,
|
| 10 |
+
"norm_pix_loss": false,
|
| 11 |
+
"num_heads": 16,
|
| 12 |
+
"patch_size": 16
|
| 13 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4846d9aa9f1d5578bcdfb53b8e6601d0e7276174c75f154836e9bcb89aa3bb53
|
| 3 |
+
size 1318206208
|