Class TrainingConfig

The fine-tuning method and hyperparameters used for fine-tuning a custom model.

TrainingConfig

[JsonConverter(typeof(TrainingConfigModelConverter))]
public class TrainingConfig

[JsonProperty(PropertyName = "earlyStoppingPatience")]
public int? EarlyStoppingPatience { get; set; }

Type	Description
int?	Stop training if the loss metric does not improve beyond 'early_stopping_threshold' for this many times of evaluation.

[JsonProperty(PropertyName = "earlyStoppingThreshold")]
public double EarlyStoppingThreshold { get; set; }

Type	Description
double	How much the loss must improve to prevent early stopping.

[JsonProperty(PropertyName = "learningRate")]
public double LearningRate { get; set; }

Type	Description
double	The initial learning rate to be used during training

[JsonProperty(PropertyName = "logModelMetricsIntervalInSteps")]
public int? LogModelMetricsIntervalInSteps { get; set; }

Type	Description
int?	Determines how frequently to log model metrics. Every step is logged for the first 20 steps and then follows this parameter for log frequency. Set to 0 to disable logging the model metrics.

[JsonProperty(PropertyName = "totalTrainingEpochs")]
public int? TotalTrainingEpochs { get; set; }

Type	Description
int?	The maximum number of training epochs to run for.

[JsonProperty(PropertyName = "trainingBatchSize")]
public int? TrainingBatchSize { get; set; }

Type	Description
int?	The batch size used during training.