Package io.github.kirstenali.deepj.optimisers

Interface ParameterOptimizer

All Known Implementing Classes:
AdamW
public interface ParameterOptimizer
Optimizer that updates a set of Parameters once per training step.

Important: implementations should treat one call to step(List) as a single optimizer step (for bias correction, schedules, etc.).

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    step(List<Parameter> params)
    Update the provided parameters (one optimizer step).

  • Method Details

    • step

      void step(List<Parameter> params)
      Update the provided parameters (one optimizer step).