VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add hyperparams (#793)

Browse Source
This commit is contained in:
Janusch Jacoby
2024-07-19 16:13:02 +02:00
committed by GitHub
parent ae903d7465
commit a7e9f0e388
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fine_tuning_job.go
View File

@@ -26,7 +26,9 @@ type FineTuningJob struct {
} }
type Hyperparameters 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 { type FineTuningJobRequest struct {

4
fine_tuning_job_test.go
View File

@@ -33,7 +33,9 @@ func TestFineTuningJob(t *testing.T) {
ValidationFile: "", ValidationFile: "",
TrainingFile: "file-abc123", TrainingFile: "file-abc123",
Hyperparameters: openai.Hyperparameters{ Hyperparameters: openai.Hyperparameters{
Epochs: "auto", Epochs: "auto",
LearningRateMultiplier: "auto",
BatchSize: "auto",
}, },
TrainedTokens: 5768, TrainedTokens: 5768,
}) })