Update README.md
Browse files
README.md
CHANGED
@@ -63,3 +63,16 @@ The following hyperparameters were used during training:
|
|
63 |
- Pytorch 2.5.0
|
64 |
- Datasets 3.0.2
|
65 |
- Tokenizers 0.20.1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
- Pytorch 2.5.0
|
64 |
- Datasets 3.0.2
|
65 |
- Tokenizers 0.20.1
|
66 |
+
|
67 |
+
### How to use
|
68 |
+
```python
|
69 |
+
# load data using `dataload.py`
|
70 |
+
|
71 |
+
model = VideoMAEForPreTraining.from_pretrained(
|
72 |
+
standardmodelbio/smb-vision-base,
|
73 |
+
trust_remote_code=True,
|
74 |
+
)
|
75 |
+
|
76 |
+
embedding = model.videomae(batch["image"])
|
77 |
+
|
78 |
+
```
|