etornam commited on
Commit
a4a5a88
·
verified ·
1 Parent(s): 4ba7359

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -16
README.md CHANGED
@@ -42,22 +42,6 @@ model.eval()
42
  embeds = model(x, is_training=False)
43
  ```
44
 
45
- You can also retrieve embeddings from the layer before the head:
46
- ```python
47
- import mlx.core as mx
48
- from mlxim.model import create_model
49
- from mlxim.io import read_rgb
50
- from mlxim.transform import ImageNetTransform
51
-
52
- transform = ImageNetTransform(train=False, img_size=224)
53
- x = transform(read_rgb("image.png"))
54
- x = mx.expand_dims(x, 0)
55
-
56
- model = create_model("vit_base_patch16_224.dinov3", num_classes=0)
57
- model.eval()
58
-
59
- embeds = model(x, is_training=False)
60
- ```
61
 
62
  ## Architecture
63
 
 
42
  embeds = model(x, is_training=False)
43
  ```
44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
 
46
  ## Architecture
47