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 |