Search code examples
tensorflowtensorflow-lite

How to embed model version or description within a tensorflow lite model file (v 1.x)

Is there a way for us to embed some text description like model version number etc into a tensorflow lite model prior to using the model on mobile devices ?

Solution

  • You can use the metadata field:

    https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/schema/schema.fbs#L903