rom1504 / embedding-reader

Efficiently read embedding in streaming from any filesystem
MIT License
94 stars 19 forks source link

add training example #2

Open rom1504 opened 2 years ago

rom1504 commented 2 years ago

for example learning a few non linear layers on top of clip embeddings

rom1504 commented 2 years ago

https://huggingface.co/datasets/imagenet-1k use this to build a linear probe layer on top of clip embeddings for imagenet