Model storage should support more frameworks/tools

tensorwerk / stockroom

Version control for software 2.0

https://tensorwerk.com

Apache License 2.0

64 stars 5 forks source link

Model storage should support more frameworks/tools #8

Open hhsecond opened 4 years ago

hhsecond commented 4 years ago

Description

stock.model's high level functions such as stock.model.load_weights currently only support PyTorch & Keras models. It needs to support other frameworks such as

Native TF (<2.0)
scikit-learn
ONNX
TBD

rlizzo commented 4 years ago

Onnx-ML would be good as well.

https://onnx.ai/

hhsecond commented 4 years ago

Thanks Rick. Added ONNX to the list