Replace scattered VRAM-based if/else chains with a declarative TOML profile system. GPU profiles (rtx3050, a100, h100, default) are selected by device name and embedded at compile time via include_str! for zero-filesystem fallback in CI/containers, with filesystem and env var overrides. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
13 lines
221 B
TOML
13 lines
221 B
TOML
# H100 PCIe/SXM (80GB VRAM) -- full production
|
|
[training]
|
|
batch_size = 1024
|
|
num_atoms = 51
|
|
buffer_size = 500_000
|
|
|
|
[experience]
|
|
gpu_n_episodes = 256
|
|
gpu_timesteps_per_episode = 500
|
|
|
|
[cuda]
|
|
cuda_stack_bytes = 65536 # 64KB
|