Method: Tensorflow::Train::Optimizer#minimize

Defined in:
lib/tensorflow/train/optimizer.rb

#minimize(loss, var_list: nil, grad_loss: nil, global_step: nil, name: nil) ⇒ Object



20
21
22
23
24
25
26
# File 'lib/tensorflow/train/optimizer.rb', line 20

def minimize(loss, var_list: nil, grad_loss: nil, global_step: nil, name: nil)
  grads_and_vars = compute_gradients(loss, var_list: var_list, grad_loss: grad_loss)
  if grads_and_vars.empty?
    raise(Error::InvalidArgumentError, "No gradients provided for any variable, check your graph for ops that do not support gradients")
  end
  apply_gradients(grads_and_vars, global_step: global_step, name: name)
end