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))
  • 相关阅读:
    Linux rcp命令详解
    Linux patch命令详解
    Linux paste命令详解
    linux od命令详解
    linux mv命令详解
    Linux more命令详解
    Linux mktemp命令
    MySQL状态变量详解
    mysql性能分析show profile/show profiles
    MySQL执行计划
  • 原文地址:https://www.cnblogs.com/Harriett-Lin/p/9594132.html
Copyright © 2011-2022 走看看