search

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

fix: log step count on the x-axis of the neptune logger. #1021

Closed RuanJohn closed 7 months ago

RuanJohn commented 7 months ago

What?

Super small fix to log step counts as the independent variable on Neptune for the evaluator. Previously we were logging step count for the actor and eval_step for the evaluator.

Logs with the fix: Screenshot from 2024-02-14 13-52-51

Logs without the fix: Screenshot from 2024-02-14 13-52-59

Link for comparing on Neptune).