我正在使用.pb
从权重和图表创建TensorFlow模型文件(tf.train.write_graph()
文件)。
我在Python {'var1': [2, 5, 20], 'var2': 'blah', 'var3': 5}
中有一个字典,其中包含我将对从网络输出的预测执行的后处理的信息。每个模型的字典中的变量都有不同的值(即,不同的模型对var1
,var2
等具有不同的值,以调整后处理算法以适用于该特定的模型)。
我的问题是:有没有办法在.pb
文件中嵌入这个字典,并在从.pb
文件加载图表时检索它以整合所有内容?否则,我必须有一个单独的数据文件,其中包含每个模型文件的这些信息,这会使事情变得麻烦。