TensorFlow实现保存训练模型为pd文件并恢复-创新互联
TensorFlow保存模型代码
上栗ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:18982081108(备注:SSL证书合作)期待与您的合作!import tensorflow as tf from tensorflow.python.framework import graph_util var1 = tf.Variable(1.0, dtype=tf.float32, name='v1') var2 = tf.Variable(2.0, dtype=tf.float32, name='v2') var3 = tf.Variable(2.0, dtype=tf.float32, name='v3') x = tf.placeholder(dtype=tf.float32, shape=None, name='x') x2 = tf.placeholder(dtype=tf.float32, shape=None, name='x2') addop = tf.add(x, x2, name='add') addop2 = tf.add(var1, var2, name='add2') addop3 = tf.add(var3, var2, name='add3') initop = tf.global_variables_initializer() model_path = './Test/model.pb' with tf.Session() as sess: sess.run(initop) print(sess.run(addop, feed_dict={x: 12, x2: 23})) output_graph_def = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['add', 'add2', 'add3']) # 将计算图写入到模型文件中 model_f = tf.gfile.FastGFile(model_path, mode="wb") model_f.write(output_graph_def.SerializeToString())
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
本文标题:TensorFlow实现保存训练模型为pd文件并恢复-创新互联
网站URL:http://azwzsj.com/article/ccppcs.html