Package ai.djl.training.tracker

Interface CyclicalTracker.ScaleFunction

Enclosing class:
CyclicalTracker
public static interface CyclicalTracker.ScaleFunction
ScaleFunction is an interface to implement a custom scale function.

  • Method Summary

    Modifier and Type
    Method
    Description
    float
    func(int steps)
    Custom scaling policy.

  • Method Details

    • func

      float func(int steps)
      Custom scaling policy. Return value has to satisfy 0<=func(steps)<=1 for all steps>1.
      Parameters:
      steps - current cycles if scaleModeCycle is true; input update number if it is false
      Returns:
      scale ratio