Downloads · 30 days
0
aniketr/mrl-convnext-tiny
mrl-convnext-tiny is a image classification model from aniketr. Use it when you need a label for an image. The card lists the license as mit.
We provide a ConvNeXt-Tiny model trained with Matryoshka Representation Learning (MRL) from scratch on ImageNet-1K, adapted from the original repository. We train with a Matryoshka Loss at [d] = {48, 96, 192, 384, 786},
Downloads · 30 days
0
Access
Public
Updated Sep 29, 2023
Repo size
470 MB
Likes
0
Public
Click a slice to open those files.
.pth470 MB · 100%
From the Hugging Face model README
We provide a ConvNeXt-Tiny model trained with Matryoshka Representation Learning (MRL) from scratch on ImageNet-1K, adapted from the original repository. We train with a Matryoshka Loss at [d] = {48, 96, 192, 384, 786},
<p align="center"> <img src="https://cdn-uploads.huggingface.co/production/uploads/64af72d4a609b29cc7b5919b/eGorc5hWdriNcEJkwP7Lz.png" alt="drawing" width="600"/> Top-1 Retrieval Accuracy (1-nearest neighbor) of mrl-convnext-tiny on ImageNet-1K with Exact search and IVF search across vector dimensionality. </p>