zoukankan      html  css  js  c++  java
  • tensorflow学习之(十一)将python代码写入文件

    #save to file
    
    import tensorflow as tf
    import  numpy as np
    
    ##(1)Save to file 把相关变量存储到文件中
    #remember to define the same dtype and shape when restore
    W = tf.Variable([[1,2,3],[3,4,5]],dtype=tf.float32,name='weights')
    b = tf.Variable([[1,2,3]],dtype=tf.float32,name='biases')
    
    init = tf.initialize_all_variables()
    saver = tf.train.Saver()
    
    with tf.Session() as sess:
        sess.run(init)
        save_path = saver.save(sess,"my_net/save_net.ckpt")
        print("Save to path : ",save_path)
    
    ##(2)restore variables 从文件中取出相关变量
    #redefine the same shape and same type for you variables
    W = tf.Variable(np.arange(6).reshape((2,3)),dtype=tf.float32,name="weights")#reshape((2,3):2行3列
    b = tf.Variable(np.arange(3).reshape((1,3)),dtype=tf.float32,name="biases")
    
    #not need init step
    saver = tf.train.Saver()
    with tf.Session() as sess:
        saver.restore(sess,"my_net/save_net.ckpt")
        print("weights: ",sess.run(W))
        print("biases: ",sess.run(b))
  • 相关阅读:
    03Qt信号与槽(2)
    01Qt中的隐式共享
    10GNU C语言函数调用
    09GNU C语言程序编译
    第一本C语言笔记(下)
    07控制器和控制卡(3)
    06控制器和控制卡(2)
    集合
    linux指令(目录类操作指令)
    面向对象三大特征
  • 原文地址:https://www.cnblogs.com/Harriett-Lin/p/9594132.html
Copyright © 2011-2022 走看看