Update llm_train_hpo_script.py

david-thrower · web-flow · commit 1953892ed60d · 2026-01-11T20:20:59.000-05:00
Add logging of parameters...
diff --git a/llm_train_hpo_script.py b/llm_train_hpo_script.py
@@ -223,8 +223,10 @@ def objective(trial):
         "PROJECTION_N": PROJECTION_N,
         "STAGE_I_B_LEARN_RATE":STAGE_I_B_LEARN_RATE
     }
+   
 
     trial_num = trial.number
+    print(f"Params for Trial {trial_num}: {params}")
 
 
     with mlflow.start_run():

Original file line number	Diff line number	Diff line change
`@@ -223,8 +223,10 @@ def objective(trial):`
`223`	`223`	`"PROJECTION_N": PROJECTION_N,`
`224`	`224`	`"STAGE_I_B_LEARN_RATE":STAGE_I_B_LEARN_RATE`
`225`	`225`	`}`
	`226`	`+`
`226`	`227`
`227`	`228`	`trial_num = trial.number`
	`229`	`+ print(f"Params for Trial {trial_num}: {params}")`
`228`	`230`
`229`	`231`
`230`	`232`	`with mlflow.start_run():`