public static class VanillaTrainingConfig.Builder extends Object
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
VanillaTrainingConfig |
build() |
VanillaTrainingConfig.Builder |
copy(VanillaTrainingConfig model) |
VanillaTrainingConfig.Builder |
earlyStoppingPatience(Integer earlyStoppingPatience) |
VanillaTrainingConfig.Builder |
earlyStoppingThreshold(Double earlyStoppingThreshold) |
VanillaTrainingConfig.Builder |
learningRate(Double learningRate) |
VanillaTrainingConfig.Builder |
logModelMetricsIntervalInSteps(Integer logModelMetricsIntervalInSteps) |
VanillaTrainingConfig.Builder |
numOfLastLayers(Integer numOfLastLayers)
The number of last layers to be fine-tuned.
|
VanillaTrainingConfig.Builder |
totalTrainingEpochs(Integer totalTrainingEpochs) |
VanillaTrainingConfig.Builder |
trainingBatchSize(Integer trainingBatchSize) |
public VanillaTrainingConfig.Builder totalTrainingEpochs(Integer totalTrainingEpochs)
public VanillaTrainingConfig.Builder learningRate(Double learningRate)
public VanillaTrainingConfig.Builder trainingBatchSize(Integer trainingBatchSize)
public VanillaTrainingConfig.Builder earlyStoppingPatience(Integer earlyStoppingPatience)
public VanillaTrainingConfig.Builder earlyStoppingThreshold(Double earlyStoppingThreshold)
public VanillaTrainingConfig.Builder logModelMetricsIntervalInSteps(Integer logModelMetricsIntervalInSteps)
public VanillaTrainingConfig.Builder numOfLastLayers(Integer numOfLastLayers)
The number of last layers to be fine-tuned.
numOfLastLayers
- the value to setpublic VanillaTrainingConfig build()
public VanillaTrainingConfig.Builder copy(VanillaTrainingConfig model)
Copyright © 2016–2024. All rights reserved.