[BUG] Crash if you set num_eval_episodes to specific values due to key splitting

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

Closed EdanToledo closed 2 months ago

EdanToledo commented 7 months ago

Title says it all. On TPUv3 try setting num_eval_episodes = 1

sash-a commented 2 months ago

Fixed by #1084