instadeepai / Mava

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

Feat/add time to json logger #962

Closed RuanJohn closed 11 months ago

RuanJohn commented 11 months ago

What?

Add elapsed wallclock time to the json files. This allows for richer plots down stream.

Extra

For now, the time is offset by the time of the first evaluation. This means that we won't take compilation time into and will have the first evaluation step at t=0. Alternatively, we could change this which would mean that the first time on the x-axis for the first evaluation step would be some non-zero value k and the gap between t=0 and t=k will be larger than the gap between all other evaluation intervals.