溫馨提示×

tensorflow如何保存變量到文件

小億
108
2024-04-03 13:01:01
欄目: 深度學習

在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,可以根據需要自定義文件路徑和文件名。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女