人工復制tensorflow書上的MNIST三部分代碼文件,分別可以歸類為:
1. 模型生成腳本
2. 訓練腳本
3. 驗證腳本
以下代碼就不放出來了,網上可以找到很多。這里就列出思路以及實際敲代碼過程中碰到的問題、疑惑等等。
模型生成腳本
如字面意思,就是專門生成訓練模型使用的腳本文件。當然首先記住定義好的所有張量名、可訓練、不可訓練張量、命名空間等等。
當然可以使用全局變量名稱方便使用。
之所以把模型生成腳本單獨作為一份腳本文件,是因為訓練腳本時所用到,在驗證腳本時也需要用到完整的模型,所以單獨成立一份腳本會比較方便。
訓練腳本
顧名思義,就是轉本用來訓練模型使用的腳本文件,使用該文件進行所有參數的初始化,指定訓練邊界以及最重要的,分步驟save當先訓練模型。
驗證腳本
驗證腳本主要用來驗證當前訓練模型在驗證集上的表現情況。使用Save類加載模型,并指定是否使用滑動平均法的影子參數(例子使用)。在加載模型到session中之前,首先還得定義模型的樣子,也就是調用模型生成腳本。并且添加驗證使用的張量以及相關計算。