instadeepai / Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX
Apache License 2.0
709 stars 83 forks source link

Make sure that config params passed to fbx.make_item_buffer are ints #1065

Closed liamclarkza closed 6 months ago

liamclarkza commented 6 months ago

What?

Casts the configuration parameters used to set up the replay buffer to be integers as currently if the configuration parameters are set as floats, such as system.buffer_size = 1e6, Mava will crash.