14 lines
231 B
TOML
14 lines
231 B
TOML
# A100 (40-80GB VRAM)
|
|
[training]
|
|
batch_size = 2048
|
|
num_atoms = 51
|
|
buffer_size = 200000
|
|
hidden_dim_base = 256
|
|
replay_buffer_vram_fraction = 0.70
|
|
|
|
[experience]
|
|
gpu_timesteps_per_episode = 500
|
|
|
|
[cuda]
|
|
cuda_stack_bytes = 65536 # 64KB
|