SAME-Project / same-project

https://sameproject.ml/
Apache License 2.0
19 stars 8 forks source link

Support keras/pytorch/other model serialisation in kubeflow backend #99

Open Bubblyworld opened 2 years ago

Bubblyworld commented 2 years ago

There are some issues with serialising keras models using dill - in particular, deserialising a Lambda layer seems to bomb. Should also check that we can serialise pytorch models, given that they're becoming a standard.