美文网首页
tensorflow

tensorflow

作者: 财务自由_lang | 来源:发表于2017-07-11 23:23 被阅读0次

    import tensorflow as tf

    import numpy as np

    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")

    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)

    w=tf.Variable(np.arane(6).reshape((2,3),dtype=tf.float32,name="weights")

    b=tf.Variable(np.arange(3),reshape((1,3)),dtype=tf.float32,name="biases")

    save=tf.train.Saver()

    with tf.Session() as sess:

         saver.restore(sess,"my_set/save_net.ckpt")

         print("weight",sess.run(w))

         print("biases",sess.run(b))

    相关文章

      网友评论

          本文标题:tensorflow

          本文链接:https://www.haomeiwen.com/subject/oowytxtx.html