第一次用TensorFlow

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

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

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

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

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

代码

参考资料

发表评论

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