mlx-vision
/

wide_resnet50_2-mlxim

Image Classification

Model card Files Files and versions

riccardomusmeci commited on Mar 20, 2024

Commit

e65b1bb

·

verified ·

1 Parent(s): d9dd3d3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ embeds = model(x)
 model = create_model("wide_resnet50_2")
 model.eval()
-embeds = model.features(x)
 ```
 ## Model Comparison

 model = create_model("wide_resnet50_2")
 model.eval()
+embeds = model.get_features(x)
 ```
 ## Model Comparison