tensorflow

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)