Interface TrainingProgressInfo.Builder

  • Method Details

    • totalStepCountPerEpoch

      TrainingProgressInfo.Builder totalStepCountPerEpoch(Long totalStepCountPerEpoch)

      The total step count per epoch.

      Parameters:
      totalStepCountPerEpoch - The total step count per epoch.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • currentStep

      TrainingProgressInfo.Builder currentStep(Long currentStep)

      The current step number.

      Parameters:
      currentStep - The current step number.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • currentEpoch

      TrainingProgressInfo.Builder currentEpoch(Long currentEpoch)

      The current epoch number.

      Parameters:
      currentEpoch - The current epoch number.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • maxEpoch

      TrainingProgressInfo.Builder maxEpoch(Long maxEpoch)

      The maximum number of epochs for this job.

      Parameters:
      maxEpoch - The maximum number of epochs for this job.
      Returns:
      Returns a reference to this object so that method calls can be chained together.