mlx-vision
/

vit_base_patch16_224.swag_lin-mlxim

Image Classification

Model card Files Files and versions Community

riccardomusmeci commited on Mar 8

Commit

685a16b

•

1 Parent(s): 9a1d8c3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ model.eval()
 logits = model(x)
 ```
-You can also use the embeds from last conv layer:
 ```python
 from mlxim.model import create_model
 from mlxim.io import read_rgb

 logits = model(x)
 ```
+You can also use the embeds from layer before head:
 ```python
 from mlxim.model import create_model
 from mlxim.io import read_rgb