f5b21406a8
Auto-scaling returned 8192 on 4GB RTX 3050 (should be 64), causing 17s/epoch instead of 0.28s. The VRAM math didn't account for IQN (1.1GB), attention, IQL, replay buffer (70% VRAM). Profile-tested values: - RTX 3050: 64 (4GB, minimal) - A100: 2048 (40-80GB) - H100: 8192 (80GB, production) - Default: 256 (conservative) Auto-scaling remains as fallback for batch_size=0 (unknown GPU). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
14 lines
248 B
TOML
14 lines
248 B
TOML
# A100 (40-80GB VRAM)
|
|
[training]
|
|
batch_size = 2048
|
|
num_atoms = 51
|
|
buffer_size = 0 # 0 = auto from VRAM
|
|
hidden_dim_base = 256
|
|
replay_buffer_vram_fraction = 0.70
|
|
|
|
[experience]
|
|
gpu_timesteps_per_episode = 500
|
|
|
|
[cuda]
|
|
cuda_stack_bytes = 65536 # 64KB
|