add hyperparams (#793)

2024-07-19 16:13:02 +02:00
parent ae903d7465
commit a7e9f0e388
2 changed files with 6 additions and 2 deletions
--- 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 {