Linux 拨号vps windows公众号手机端

tensorflow如何保存变量到文件

lewis 6年前 (2020-01-22) 阅读数 11 #大数据
文章标签 tensorflow

在TensorFlow中,可以使用tf.train.Saver()来保存变量到文件中。以下是一个保存变量的示例代码:

import tensorflow as tf

# 创建变量
weights = tf.Variable(tf.random_normal([10, 10]), name='weights')
biases = tf.Variable(tf.zeros([10]), name='biases')

# 初始化变量
init = tf.global_variables_initializer()

# 创建Saver对象
saver = tf.train.Saver()

# 启动Session
with tf.Session() as sess:
    sess.run(init)
    
    # 将变量保存到文件中
    saver.save(sess, 'model/model.ckpt')

在上面的代码中,首先创建了两个变量weightsbiases,然后初始化这些变量,并创建了一个Saver对象。最后在Session中使用Saver对象的save()方法将变量保存到文件中。保存的文件名为model.ckpt,可以根据需要自定义文件路径和文件名。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门