[BUG] Crash if you set num_eval_episodes to specific values due to key splitting
EdanToledo opened this issue · comments
Title says it all. On TPUv3 try setting num_eval_episodes = 1
instadeepai / Mava
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX