关于动量梯度下降

来源:3-4 卷积神经网络实战

技术进步官

2020-04-20

这一节是不是还没有运用到动量梯度下降,还是说train_op里面帮我们实现了呢?

写回答

1回答

正十七

2020-04-21

这一节中我们使用的是Adam梯度下降算法:

train_op = tf.train.AdamOptimizer(1e-3).minimize(loss)

如果你想使用momentum的话,把这行代码改掉即可。

1
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程