Tensorflow加載預訓練模型和保存模型


1. Tensorflow模型文件

(1)checkpoint
該文件是文本文件,里面記錄了保存的最新的checkpoint文件以及其他checkpoint文件列表。在測試的時候,可以通過修改這個文件,指定具體使用哪個模型
(2)meta文件
這個文件保存的是計算圖結構,可以理解為神經網絡結構圖。是一個二進制的pb格式文件,包含變量、op、集合等。
(3)ckpt文件
ckpt文件是二進制文件,保存了所有的weights、biases、gradients等變量。在tensorflow 0.11之前,保存在.ckpt文件中。0.11后,通過兩個文件保存

  • 一個是data文件:保存了所有的變量
  • 一個是index文件:保存了所有變量(包括權重、偏置、梯度等的值)是一個key-value的字典。

2. 保存Tensorflow模型

參考


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM