optimizer = tf.train.GradientDescentOptimizer(0.001)
train = optimizer.minimize(loss)
第一行设置了训练的步长为0.001
第二行设置了训练的目标是使损失值最小化
tenserflow设置学习步长
- 上一篇：tenserflow损失函数的设置
- 下一篇：tf.estimator的使用
相关内容
发表评论
更多
网友评论 （0 条评论）
暂无评论