Class Adam.Builder

    • Method Detail

      • optLearningRateTracker

        public Adam.Builder optLearningRateTracker​(Tracker learningRateTracker)
        Sets the Tracker for this optimizer.
        Parameters:
        learningRateTracker - the Tracker to be set
        Returns:
        this Builder
      • optBeta1

        public Adam.Builder optBeta1​(float beta1)
        Sets the decay rate for the first moment estimates.
        Parameters:
        beta1 - the deacay rate for the the first moment estimates
        Returns:
        this Builder
      • optBeta2

        public Adam.Builder optBeta2​(float beta2)
        Sets the decay rate for the second moment estimates.
        Parameters:
        beta2 - the decay rate for the the second moment estimates
        Returns:
        this Builder
      • optEpsilon

        public Adam.Builder optEpsilon​(float epsilon)
        Sets \(epsilon\) - a small quantity for numerical stability.
        Parameters:
        epsilon - a small quantity for numerical stability
        Returns:
        this Builder
      • build

        public Adam build()
        Builds a Adam block.
        Returns:
        the Adam block