缥缈电脑

文章
7
资源
0
加入时间
2年10月21天

tensorflow保存图和权重

有时候,我们需要将TensorFlow的模型导出为单个文件(同时包含模型架构定义与权重),方便在其他地方使用(如在c++中部署网络)。利用tf.train.write_graph()默认情况下只导出了网络的定义(没有权重),而利用tf.train.Saver().save()导出的文件graph_def与权重是分离的,因此需要采用别的方法。我们知道,graph_def文件中没有包含网络中的Va