search

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data
https://rikai.readthedocs.io/en/latest/
Apache License 2.0
138 stars 19 forks source link

Create a model type for feature extractor #663

Closed eddyxu closed 2 years ago

eddyxu commented 2 years ago

https://pytorch.org/vision/stable/feature_extraction.html

We can use this model to extract embeddings from a pytorch model