欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例

 更新時(shí)間：2020年02月10日 11:16:45   作者：奔跑的Yancy  
今天小編就為大家分享一篇TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

在使用TensorFlow的過(guò)程中，保存模型參數(shù)變量是很重要的一個(gè)環(huán)節(jié)，既可以保證訓(xùn)練過(guò)程信息不丟失，也可以幫助我們?cè)谛枰焖倩謴?fù)或使用一個(gè)模型的時(shí)候，利用之前保存好的參數(shù)之間導(dǎo)入，可以節(jié)省大量的訓(xùn)練時(shí)間。本文通過(guò)最簡(jiǎn)單的例程教大家如何保存和讀取.ckpt文件。

一、保存到文件

首先是導(dǎo)入必要的東西：

import tensorflow as tf
import numpy as np

隨便寫(xiě)幾個(gè)變量：

# Save to file
# remember to define the same dtype and shape when restore
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.initialize_all_variables()

定義一個(gè)saver，來(lái)存儲(chǔ)我們的各種變量：

saver = tf.train.Saver()

保存的文件用.ckpt后綴：

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

上面我們就完成了保存操作。

接下來(lái)我們要把之前保存過(guò)的變量取出來(lái)。

二、取出之前保存的變量

這里要注意，取出時(shí)要先開(kāi)辟一個(gè)容器來(lái)裝，shape和type要和我們之前保存的.ckpt一樣。

# restore variables
# redefine the same shape and same type for your variables
W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")
b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")

restore時(shí)，不需要進(jìn)行init= tf.initialize_all_variables()操作。

利用saver提取文件：

saver = tf.train.Saver()
with tf.Session() as sess:
  saver.restore(sess, "my_net/save_net.ckpt")
  print("weights:", sess.run(W))
  print("biases:", sess.run(b))

結(jié)果：

以上這篇TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論