gahow/aituner
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix pilot burn-in selection manifest

Browse Source
This commit is contained in:
Gahow Wang
2026-07-14 12:52:00 +08:00
parent 16239bef00
commit 8eeba597b3
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
runs/fidelity-headroom/prepare_pilot.py
View File

@@ -227,7 +227,11 @@ def build_manifest(
for level, multiplier in TARGET_MULTIPLIERS.items(): for level, multiplier in TARGET_MULTIPLIERS.items():
target_rate = float(config["frontier_req_s_gpu"]) * multiplier target_rate = float(config["frontier_req_s_gpu"]) * multiplier
target_count = round(target_rate * durations["low1"] * tp) target_count = round(target_rate * durations["low1"] * tp)
roles = [role for role in ROLES if role == "burnin" or role.startswith(level)] roles = [
role
for role in ROLES
if role.startswith(level) or (level == "low" and role == "burnin")
]
selections = {} selections = {}
for role in roles: for role in roles:
anchor, selected = attainable_anchor(loaded[role][tier], target_count) anchor, selected = attainable_anchor(loaded[role][tier], target_count)