1. 程式人生 > >python神經網路(五)輸入手寫數字進行識別

python神經網路(五)輸入手寫數字進行識別

一、斷點續訓
為防止突然斷電、引數白跑的情況發生,在backward中加入類似於之前test中載入ckpt的操作,給所有w和b賦儲存在ckpt中的值:

(1) 如果儲存斷點檔案的目錄資料夾中,包含有效斷點狀態檔案,則返回該檔案:

  • 引數說明
    checkpoint_dir: 表示儲存斷點檔案的目錄
    latest_filename: 斷點檔案的可選名稱,預設為checkpoint
ckpt = tf.train.get_checkpoint_state(checkpoint_dir,\
 latest_filename = None)

**(2)**如果ckpt存在,且儲存的模型在指定路徑中存在

	if ckpt and ckpt.model_checkpoint_path: 

恢復當前會話,將ckpt中的值賦給 w 和 b
引數說明:sess:表示當前會話,之前儲存的結果會被載入入這個會話
ckpt.model_checkpoint_path:表示模型儲存的位置,不需要提供模型的名字,
因為有了位置會自動去檢視checkpoint檔案,看最新的模型叫什麼

【注】內容來自mooc人工智慧實踐第六講