aniketr commited on
Commit
531993d
1 Parent(s): df77548

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
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>