Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,17 @@
|
|
1 |
---
|
2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: mit
|
3 |
+
datasets:
|
4 |
+
- imagenet-1k
|
5 |
+
metrics:
|
6 |
+
- accuracy
|
7 |
+
pipeline_tag: image-classification
|
8 |
---
|
9 |
+
# mrl-convnext-tiny
|
10 |
+
|
11 |
+
We provide a ConvNeXt-Tiny model trained with [Matryoshka Representation Learning](https://github.com/RAIVNLab/MRL) (MRL) from scratch on ImageNet-1K,
|
12 |
+
adapted from [the original repository](https://github.com/facebookresearch/ConvNeXt). We train with a Matryoshka Loss at [d] = {48, 96, 192, 384, 786},
|
13 |
+
|
14 |
+
<p align="center">
|
15 |
+
<img src="https://cdn-uploads.huggingface.co/production/uploads/64af72d4a609b29cc7b5919b/eGorc5hWdriNcEJkwP7Lz.png" alt="drawing" width="600"/>
|
16 |
+
Top-1 Retrieval Accuracy (1-nearest neighbor) of mrl-convnext-tiny on ImageNet-1K with Exact search and IVF search across vector dimensionality.
|
17 |
+
</p>
|