gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

perf: allow num_atoms up to 101 — H100 handles larger C51 distributions

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-03-22 01:42:38 +01:00
parent e464f89a04
commit 109f4e9414
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
crates/ml/src/hyperopt/adapters/dqn.rs
View File

@@ -486,7 +486,7 @@ impl ParameterSpace for DQNParams {
(0.1_f64.ln(), 1.0_f64.ln()), // 12: noisy_sigma_init (log scale)
(128.0, 512.0), // 13: dueling_hidden_dim (linear, step=128)
(3.0, 5.0), // 14: n_steps (raised min: n=3 is Rainbow standard)
(11.0, 51.0), // 15: num_atoms (11=RTX3050, 51=H100 — GPU profile caps this)
(11.0, 101.0), // 15: num_atoms (11=RTX3050, 51=H100 default, 101=max — C51 resolution)
// Weight decay (1D)
(1e-4_f64.ln(), 1e-2_f64.ln()), // 16: weight_decay (C2 FIX: 10x stronger — prevents 14x val/train overfitting)