MNIST后续

2018-09-18  本文已影响0人  __main__

人工复制tensorflow书上的MNIST三部分代码文件,分别可以归类为:

1. 模型生成脚本
2. 训练脚本
3. 验证脚本

以下代码就不放出来了,网上可以找到很多。这里就列出思路以及实际敲代码过程中碰到的问题、疑惑等等。


模型生成脚本

如字面意思,就是专门生成训练模型使用的脚本文件。当然首先记住定义好的所有张量名、可训练、不可训练张量、命名空间等等。
当然可以使用全局变量名称方便使用。

之所以把模型生成脚本单独作为一份脚本文件,是因为训练脚本时所用到,在验证脚本时也需要用到完整的模型,所以单独成立一份脚本会比较方便。

训练脚本

顾名思义,就是转本用来训练模型使用的脚本文件,使用该文件进行所有参数的初始化,指定训练边界以及最重要的,分步骤save当先训练模型。

验证脚本

验证脚本主要用来验证当前训练模型在验证集上的表现情况。使用Save类加载模型,并指定是否使用滑动平均法的影子参数(例子使用)。在加载模型到session中之前,首先还得定义模型的样子,也就是调用模型生成脚本。并且添加验证使用的张量以及相关计算。

上一篇下一篇

猜你喜欢

热点阅读