unilab.ipc.memory_budget.estimate_offpolicy_bytes

unilab.ipc.memory_budget.estimate_offpolicy_bytes(num_envs, replay_buffer_n, obs_dim, action_dim, critic_dim, batch_size, updates_per_step)[source]

Estimate memory for off-policy replay buffer + double-buffer slots.

Parameters:

  • num_envs (int)

  • replay_buffer_n (int)

  • obs_dim (int)

  • action_dim (int)

  • critic_dim (int)

  • batch_size (int)

  • updates_per_step (int)

Return type:

dict[str, int | str]