Skip to content

Commit b64f47f

Browse files
Update llm_train_hpo_script.py
Trigger a scale down test...
1 parent 10ef27b commit b64f47f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

llm_train_hpo_script.py

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

1414
DATE_0 = "2026-01-08"
1515
EXPERIMENT_NAME = "stacked-transformer-mmc"
16-
SAMPLE_SIZE_i_a = 300
17-
SAMPLE_SIZE_i_b = 600
16+
SAMPLE_SIZE_i_a = 10 # 300
17+
SAMPLE_SIZE_i_b = 10 # 600
1818
DATA_SET_NAME = "WEB-Bible"
1919
ADDITIONAL_METADATA = "2-levels"
2020
EXPERIMENT_ITERATION = "0001"
@@ -87,7 +87,7 @@ def objective(trial):
8787
PHASE_I_B_SAMPLES_TO_CREATE = 20
8888
PHASE_I_B_VAL_SPLIT = 0.15
8989

90-
PHASE_I_B_SAMPLE_EXPANSION_BATCH_SIZE = 100
90+
PHASE_I_B_SAMPLE_EXPANSION_BATCH_SIZE = 10 # 100
9191

9292
PROMPT_LENGTH = 1
9393
MAX_SEQ_LENGTH = 40

0 commit comments

Comments
 (0)