add hyperparams (#793)
This commit is contained in:
@@ -27,6 +27,8 @@ type FineTuningJob struct {
|
||||
|
||||
type Hyperparameters struct {
|
||||
Epochs any `json:"n_epochs,omitempty"`
|
||||
LearningRateMultiplier any `json:"learning_rate_multiplier,omitempty"`
|
||||
BatchSize any `json:"batch_size,omitempty"`
|
||||
}
|
||||
|
||||
type FineTuningJobRequest struct {
|
||||
|
||||
@@ -34,6 +34,8 @@ func TestFineTuningJob(t *testing.T) {
|
||||
TrainingFile: "file-abc123",
|
||||
Hyperparameters: openai.Hyperparameters{
|
||||
Epochs: "auto",
|
||||
LearningRateMultiplier: "auto",
|
||||
BatchSize: "auto",
|
||||
},
|
||||
TrainedTokens: 5768,
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user