8a68bdf21d
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
235 B
TOML
13 lines
235 B
TOML
# Default GPU profile -- conservative settings for unknown GPUs
|
|
[training]
|
|
batch_size = 256
|
|
num_atoms = 21
|
|
buffer_size = 50_000
|
|
|
|
[experience]
|
|
gpu_n_episodes = 64
|
|
gpu_timesteps_per_episode = 200
|
|
|
|
[cuda]
|
|
cuda_stack_bytes = 32768 # 32KB
|