chenz53 commited on
Commit
16e2d73
1 Parent(s): 0425418

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -0
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
+ ```