add hyperparams (#793)

janukobytsch · web-flow · commit a7e9f0e3880d · 2024-07-19T15:13:02.000+01:00
diff --git a/fine_tuning_job.go b/fine_tuning_job.go
@@ -26,7 +26,9 @@ type FineTuningJob struct {
 }
 
 type Hyperparameters struct {
-	Epochs any `json:"n_epochs,omitempty"`
+	Epochs                 any `json:"n_epochs,omitempty"`
+	LearningRateMultiplier any `json:"learning_rate_multiplier,omitempty"`
+	BatchSize              any `json:"batch_size,omitempty"`
 }
 
 type FineTuningJobRequest struct {
diff --git a/fine_tuning_job_test.go b/fine_tuning_job_test.go
@@ -33,7 +33,9 @@ func TestFineTuningJob(t *testing.T) {
 				ValidationFile: "",
 				TrainingFile:   "file-abc123",
 				Hyperparameters: openai.Hyperparameters{
-					Epochs: "auto",
+					Epochs:                 "auto",
+					LearningRateMultiplier: "auto",
+					BatchSize:              "auto",
 				},
 				TrainedTokens: 5768,
 			})

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,9 @@ type FineTuningJob struct {`
`26`	`26`	`}`
`27`	`27`
`28`	`28`	`type Hyperparameters struct {`
`29`		- Epochs any `json:"n_epochs,omitempty"`
	`29`	+ Epochs any `json:"n_epochs,omitempty"`
	`30`	+ LearningRateMultiplier any `json:"learning_rate_multiplier,omitempty"`
	`31`	+ BatchSize any `json:"batch_size,omitempty"`
`30`	`32`	`}`
`31`	`33`
`32`	`34`	`type FineTuningJobRequest struct {`