据我所知,tf.train.Saver()
接口只能将文件前缀作为输入,tensorflow本身可以直接处理模型到磁盘的序列化。
我有兴趣将tensorflow
模型的当前状态写入PostgreSQL数据库,并且一直在研究将tensorflow
对象的当前状态序列化到内存中的某个位置,这样我就可以将该序列化状态保存到PostgreSQL BYTEA
列。我该怎么做呢?
我当前的解决方法是允许tf.train.Saver()
在tempfile
位置写入磁盘,读取并序列化这些文件的内容,然后重写到这些文件位置并允许tf.train.Saver()
从他们那里读取以在推理时加载图表。