今天我們來介紹一下TensorFlow常會使用到的幾種模組。
1. tf.train.Checkpoint : 變數的保存與還原
tf.train.Checkpoint() 能夠接受的初始化參數比較特殊,是一個** kwargs。也就是一系列的鍵值對,鍵名沒有限制,值為一個需要保護得對象。
模型訓練完成需要保存時可以使用:
(save_path_with_prefix
是儲存檔案的目錄+前綴)
還原模型變數:
(save_path_with_prefix_and_index
是之前儲存文件的目錄+前綴+編號)
2. TensorBoard : 訓練過程可視化
在代碼目錄下建立一個名為tensorboard的資料夾來存放TensorBoard的紀錄文件,並實例化一個紀錄器。
當需要紀錄訓練過程中的參數時 :
(每運行一次tf.summary.scalar()
,紀錄器就會像紀錄檔案中寫入一條紀錄。)
當需要對訓練過程可視化時,在代碼目錄打開終端:
(如若需要可進入TensorFlow的conda環境
)
(圖片擷取至網路)
(可透過右上方按鈕更新資訊,預設是30秒自動更新一次資料)
剩下還有幾種,我們明天繼續來介紹~