Tensorflow 2.0:已保存模型中的变量

时间:2019-04-07 19:43:58

标签: tensorflow tensorflow2.0

saveed_model API是否以协议缓冲区(pb)格式保存了变量文件?如果没有,有没有办法在不使用tensorflow API的情况下加载此文件(还原/加载)

3 个答案:

答案 0 :(得分:1)

有一个纯Python API,如果有帮助,它不使用TensorFlow操作:list variablesload a single variable。对于SavedModel,您可以将其指向变量/子目录。

还有TensorBundle,它是C ++中的实现。

如果这些都不有用,答案可能是“否”。从理论上讲,它可以分解成一个单独的包装;如果您对此感兴趣,可以随时与我们联系。

答案 1 :(得分:0)

您可以使用tf.keras.models.load_model从save_model加载模型,而得到的是一个tf.keras.Model对象。

答案 2 :(得分:-1)

我不确定是否已通过验证。但是似乎将List_variables和load_variable指向SavedModel的变量子目录不起作用。我们将看到“检查点”文件未命中断言。 WA将创建一个检查点文件,其中一行指向数据文件名。 model_checkpoint_path:“变量”