Support model type in model spec

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

Support model type in model spec #525

Closed eddyxu closed 2 years ago

eddyxu commented 2 years ago

This PR sets a rule to find a "ModelType" class via MODEL_TYPE attribute. It points to a class "ModelSpec" which wraps the actual model (pytorch), and acts as a proxy. It offers the flexibility to adapt non-standard pytorch models.

eddyxu commented 2 years ago

Related to https://github.com/eto-ai/rikai/issues/520 and support rikai-yolov5 using pytorch runner

eddyxu commented 2 years ago

Ok, filed #533 and #532 as follow ups