unilab.ipc.memory_budget.estimate_appo_bytes
-
unilab.ipc.memory_budget.estimate_appo_bytes(num_envs, steps_per_env, obs_dim, action_dim, critic_dim, num_slots=4)[source]
Estimate memory for APPO rollout ring buffer.
- Parameters:
num_envs (int)
steps_per_env (int)
obs_dim (int)
action_dim (int)
critic_dim (int)
num_slots (int)
- Return type:
dict[str, int | str]