Skip to content

Commit a54390a

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Add better metadata.
1 parent 58cc7db commit a54390a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,14 +13,17 @@
1313
print(answer.stdout)
1414

1515

16-
EXPERIMENT_ITERATION = "0002"
16+
BASE_EXPERIMENT_NAME = "247-2025-09-27--hpo-study-with-sampling-plus-penalties"
17+
EXPERIMENT_ITERATION = "0001"
18+
EXPERIMENT_SCALE = "single-worker"
19+
SAMPLES_N = "75"
1720

1821
N_TRIALS = 50
1922

2023

2124
mlflow.set_tracking_uri(uri=f"http://127.0.0.1:{MLFLOW_PORT}")
2225

23-
mlflow.set_experiment(f"single-worker-1st-pass-tuning-{EXPERIMENT_ITERATION}-a")
26+
mlflow.set_experiment(f"{BASE_EXPERIMENT_NAME}--{SAMPLES_N}--{EXPERIMENT_SCALE}--{EXPERIMENT_ITERATION}-a")
2427

2528

2629

@@ -60,7 +63,7 @@ def objective(trial: optuna.Trial) -> float:
6063
# Number of text samples to create: # Number of text samples (of approximately max_seq_len) to create
6164
# Raises RAM in a linear fashion
6265

63-
SAMPLES_TO_CREATE = 10
66+
SAMPLES_TO_CREATE = 75
6467

6568
# How many tokens to provide before expecting the next token to be predicted.
6669
# Half this = double RAM (inversely proportional to RAM requirement)

0 commit comments

Comments
 (0)