将其他变量写入TensorFlow .pb文件

时间:2017-04-08 17:30:33

标签: python dictionary tensorflow

我正在使用.pb从权重和图表创建TensorFlow模型文件(tf.train.write_graph()文件)。

我在Python {'var1': [2, 5, 20], 'var2': 'blah', 'var3': 5}中有一个字典,其中包含我将对从网络输出的预测执行的后处理的信息。每个模型的字典中的变量都有不同的值(即,不同的模型对var1var2等具有不同的值,以调整后处理算法以适用于该特定的模型)。

我的问题是:有没有办法在.pb文件中嵌入这个字典,并在从.pb文件加载图表时检索它以整合所有内容?否则,我必须有一个单独的数据文件,其中包含每个模型文件的这些信息,这会使事情变得麻烦。

0 个答案:

没有答案