Support saving models via tf.saved_models

secondmind-labs / trieste

A Bayesian optimization toolbox built on TensorFlow

Apache License 2.0

221 stars 42 forks source link

Support saving models via tf.saved_models #725

Closed uri-granta closed 1 year ago

uri-granta commented 1 year ago

It's currently difficult to export trieste model wrappers using tf.saved_models, in part because they don't inherit from tf.Module. We should investigate how to do this as cleanly as possible.

uri-granta commented 1 year ago

Implemented in https://github.com/secondmind-labs/trieste/pull/750