instadeepai / Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

Home Page:https://id-mava.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[MAINTAIN] Making the best checkpoint and the absolute metric support norm params

OmaymaMahjoub opened this issue · comments

Please describe what needs to be maintained?

Currently the best checkpointer and the absolute metric features are working fine in the default case, however, they both don't support the case of the normalization of the params such as normalize observation and normalize the target value.

Describe the outcome you'd like

Create the option of storing norm_params in the best checkpointer component