Class EarlyStoppingListener.EarlyStoppedException

java.lang.Object
java.lang.Throwable
java.lang.Exception
java.lang.RuntimeException
ai.djl.training.listener.EarlyStoppingListener.EarlyStoppedException
All Implemented Interfaces:
Serializable
Enclosing class:
EarlyStoppingListener

public static class EarlyStoppingListener.EarlyStoppedException extends RuntimeException
Thrown when training is stopped early, the message will contain the reason why it is stopped early.
See Also:
  • Constructor Details

    • EarlyStoppedException

      public EarlyStoppedException(int stopEpoch, String message)
      Constructs an EarlyStoppingListener.EarlyStoppedException with the specified message and epoch.
      Parameters:
      stopEpoch - the epoch at which training was stopped early
      message - the message/reason why training was stopped early
  • Method Details

    • getStopEpoch

      public int getStopEpoch()
      Gets the epoch at which training was stopped early.
      Returns:
      the epoch at which training was stopped early.