tf.train.GradientDescentOptimizer 后期专门补上

__init__(
    learning_rate,
    use_locking=False,
    name='GradientDescent'
)

他是一个类,下面有好多方法,这里讲一下,最后代码用到的 minimize

minize

minimize(
    loss,
    global_step=None,
    var_list=None,
    gate_gradients=GATE_OP,
    aggregation_method=None,
    colocate_gradients_with_ops=False,
    name=None,
    grad_loss=None
)

说明

返回的是一个操作(op),通过更新 var_list 来最小化loss

猜你喜欢

转载自blog.csdn.net/qq_32806793/article/details/85009929