国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 編程 > Python > 正文

TensorFLow用Saver保存和恢復(fù)變量

2020-01-04 15:43:31
字體:
供稿:網(wǎng)友

本文為大家分享了TensorFLow用Saver保存和恢復(fù)變量的具體代碼,供大家參考,具體內(nèi)容如下

建立文件tensor_save.py, 保存變量v1,v2的tensor到checkpoint files中,名稱分別設(shè)置為v3,v4。

import tensorflow as tf# Create some variables.v1 = tf.Variable(3, name="v1")v2 = tf.Variable(4, name="v2")# Create modely=tf.add(v1,v2)# Add an op to initialize the variables.init_op = tf.initialize_all_variables()# Add ops to save and restore all the variables.saver = tf.train.Saver({'v3':v1,'v4':v2})# Later, launch the model, initialize the variables, do some work, save the# variables to disk.with tf.Session() as sess: sess.run(init_op) print("v1 = ", v1.eval()) print("v2 = ", v2.eval()) # Save the variables to disk. save_path = saver.save(sess, "f:/tmp/model.ckpt") print ("Model saved in file: ", save_path)

建立文件tensor_restror.py, 將checkpoint files中名稱分別為v3,v4的tensor分別恢復(fù)到變量v3,v4中。

import tensorflow as tf# Create some variables.v3 = tf.Variable(0, name="v3")v4 = tf.Variable(0, name="v4")# Create modely=tf.mul(v3,v4)# Add ops to save and restore all the variables.saver = tf.train.Saver()# Later, launch the model, use the saver to restore variables from disk, and# do some work with the model.with tf.Session() as sess: # Restore variables from disk. saver.restore(sess, "f:/tmp/model.ckpt") print ("Model restored.") print ("v3 = ", v3.eval()) print ("v4 = ", v4.eval()) print ("y = ",sess.run(y))

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持VEVB武林網(wǎng)。



注:相關(guān)教程知識閱讀請移步到python教程頻道。
發(fā)表評論 共有條評論
用戶名: 密碼:
驗(yàn)證碼: 匿名發(fā)表
主站蜘蛛池模板: 南丰县| 抚远县| 梁山县| 阳新县| 开远市| 宽甸| 凌云县| 临漳县| 辽宁省| 亳州市| 米脂县| 保康县| 会东县| 永胜县| 道真| 财经| 沽源县| 宁海县| 南郑县| 佛教| 云林县| 凌云县| 德安县| 沅江市| 息烽县| 安福县| 宜兰市| 湖南省| 弋阳县| 略阳县| 辽源市| 东安县| 富宁县| 唐山市| 伊宁县| 南乐县| 防城港市| 惠来县| 开阳县| 长阳| 冕宁县|