tensorflow如何保存变量到文件

avatar
作者
筋斗云
阅读量:0

在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,可以根据需要自定义文件路径和文件名。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!