Mikhail Karasikov commited on
Commit
d43e663
·
1 Parent(s): bc68d96

added usage

Browse files
Files changed (1) hide show
  1. README.md +16 -0
README.md CHANGED
@@ -1,3 +1,19 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+
6
+ ### Usage
7
+
8
+ ```python
9
+ import timm
10
+ from timm.data import resolve_data_config
11
+ from timm.data.transforms_factory import create_transform
12
+ from huggingface_hub import login
13
+
14
+ login() # login or use an access token
15
+
16
+ model = timm.create_model("hf-hub:kaiko-ai/midnight", pretrained=True)
17
+ transform = create_transform(**resolve_data_config(model.pretrained_cfg, model=model))
18
+ model.eval()
19
+ ```