Update README.md
Browse files
README.md
CHANGED
|
@@ -60,7 +60,7 @@ embeds = model(x)
|
|
| 60 |
model = create_model("resnet34")
|
| 61 |
model.eval()
|
| 62 |
|
| 63 |
-
embeds = model.
|
| 64 |
```
|
| 65 |
|
| 66 |
## Model Comparison
|
|
|
|
| 60 |
model = create_model("resnet34")
|
| 61 |
model.eval()
|
| 62 |
|
| 63 |
+
embeds = model.get_features(x)
|
| 64 |
```
|
| 65 |
|
| 66 |
## Model Comparison
|