-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy paththinkprm_sequential_scaling.yaml
More file actions
21 lines (21 loc) · 1.09 KB
/
Copy paththinkprm_sequential_scaling.yaml
File metadata and controls
21 lines (21 loc) · 1.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# refer to src/sal/config.py for more options
filter_duplicates: true
approach: beam_search
n: 16
prm_batch_size: 4
search_batch_size: 1
dataset_start: 0
dataset_end: 100
seed: 42
model_path: Meta-Llama/Llama-3.2-3B-Instruct
output_dir: outputs/results/test-time-scaling/MATH-100/beam-search/Llama-3.2-3B-Instruct/thinkprm/
temperature: .6
prm_temperature: 0.0
prm_n: 1
prm_type: thinkprm
verifier_endpoint: "http://127.0.0.1:31111"
n_thinking_rounds: 1
system_prompt: "Solve the following math problem efficiently and clearly:\n\n- For simple problems (2 steps or fewer):\nProvide a concise solution with minimal explanation.\n\n- For complex problems (3 steps or more):\nUse this step-by-step format:\n\n## Step 1: [Concise description]\n[Brief explanation and calculations]\n\n## Step 2: [Concise description]\n[Brief explanation and calculations]\n\n...\n\nRegardless of the approach, always conclude with:\n\nTherefore, the final answer is: $\\boxed{answer}$. I hope it is correct.\n\nWhere [answer] is just the final number or expression that solves the problem."
max_tokens: 4096
num_iterations: 20
beam_width: 4