有效地将tensorflow图/会话序列化为db?

时间:2017-07-25 15:59:23

标签: python serialization machine-learning tensorflow

据我所知,tf.train.Saver()接口只能将文件前缀作为输入,tensorflow本身可以直接处理模型到磁盘的序列化。

我有兴趣将tensorflow模型的当前状态写入PostgreSQL数据库,并且一直在研究将tensorflow对象的当前状态序列化到内存中的某个位置,这样我就可以将该序列化状态保存到PostgreSQL BYTEA列。我该怎么做呢?

我当前的解决方法是允许tf.train.Saver()tempfile位置写入磁盘,读取并序列化这些文件的内容,然后重写到这些文件位置并允许tf.train.Saver()从他们那里读取以在推理时加载图表。

0 个答案:

没有答案