- 构建 graph
现在可以开始构建graph了。这里我们先将构建graph需要的组件以函数的形式写出来,后面组合这些组件,只需调用这些函数即可。这样可以使代码的结构清晰,以方便后期调整参数,调试模型。
首先建立placeholder,用来存储输入数据X和Y,以及在优化中会用到的Dropout参数keep\_prob。这些占位符可以在执行session时再输入具体信息,在后期有调整的余地。
LeNet5模型的结构如下:
在构建网络前,先初始化参数(LeNet5中的参数存在于两个conv2d层和3个全连接层,这里只需初始化W1和W2,全连接层的参数会在运算时自动被建立和初始化;另外,偏置参数b也无须初始化):
第7章 深度学习
209