Skip to content

Commit 7040e27

Browse files
Update llm_train_hpo_script.py
Fix broken parameter flow...
1 parent 43244d1 commit 7040e27

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

llm_train_hpo_script.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,16 @@
1313

1414
DATE_0 = "2026-01-09"
1515
EXPERIMENT_NAME = "stacked-transformer-mmc"
16-
SAMPLE_SIZE_i_a = 300
17-
SAMPLE_SIZE_i_b = 600
16+
17+
PHASE_I_A_SAMPLES_TO_CREATE = 300
18+
# Samples to use for the main training stage
19+
PHASE_I_B_SAMPLES_TO_CREATE = 600
20+
1821
DATA_SET_NAME = "WEB-Bible"
1922
ADDITIONAL_METADATA = "2-4-levels"
2023
EXPERIMENT_ITERATION = "0003"
2124
EXPERIMENT_NAME = f"{DATE_0}-{EXPERIMENT_NAME}--{ADDITIONAL_METADATA}--{DATA_SET_NAME}-" +\
22-
f"ia-{SAMPLE_SIZE_i_a}-ib-{SAMPLE_SIZE_i_b}--{EXPERIMENT_ITERATION}-a"
25+
f"ia-{PHASE_I_A_SAMPLES_TO_CREATE}-ib-{PHASE_I_B_SAMPLES_TO_CREATE}--{EXPERIMENT_ITERATION}-a"
2326

2427
N_TRIALS = 40
2528

0 commit comments

Comments
 (0)