def optimize(self,
loss: tf.Tensor,
learning_rate: float) -> tf.Operation:
with tf.variable_scope("optimize"):
optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate)
return optimizer.minimize(loss, name="train_op")
文章目录