tensorflow基本教程9:保存变量值

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Dian1pei2xiao3/article/details/82290675

import tensorflow as tf
import numpy as np

##Save to file
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.tables_initializer()

with tf.Session()as sess:

    sess.run(init)
    saver=tf.train.Saver()
    save_path=saver.save(sess,"my_net/save_net.ckpt")
    print("Save to path:",save_path)

猜你喜欢

转载自blog.csdn.net/Dian1pei2xiao3/article/details/82290675
今日推荐