Interface Optimizer

public interface Optimizer
Optimizer interface for training neural networks.
Since:
2.0
Author:
Silvere Martin-Michiellot, Gemini AI (Google DeepMind)

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    reset()
    Resets internal state (e.g. momentum buffers).
    void
    update(Tensor<?> params, Tensor<?> grads)
    Updates the improved parameters based on gradients.

  • Method Details

    • update

      void update(Tensor<?> params, Tensor<?> grads)
      Updates the improved parameters based on gradients.
      Parameters:
      params - parameters to update.
      grads - gradients corresponding to parameters.

    • reset

      void reset()
      Resets internal state (e.g. momentum buffers).