第一次用 TensorFlow

3 月份的时候其实用过一次 TF 的,但当时记忆一点都不深……而且没有硬件,跑什么东西都跑不起来。

现在 TF 出了正式版了,改动也比较大,于是就重新学 TF。

下面是一个 LeNet5 的例子,是根据 Tutorial 改的。

TensorFlow Tutorial 里面 “MNIST 高级教程” 里面是有一个 LeNet5 的例子的。跑完之后看了一下 Yan Lecun 的文章,发现好像教程和文章并不完全对应。教程中的每层的特征数更多,并且加入了 Dropout。那么就自己对着论文试着写一下这个网络吧。(反正肯定还跟作者文章中的网络不对应)

首先,输入改为了 28x28。之后,里面的特征维度按照 Lecon 的文章改了一下。

代码

参考资料

发表评论

电子邮件地址不会被公开。 必填项已用*标注