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

Feat/store json to neptune #997

Closed RuanJohn closed 8 months ago

RuanJohn commented 8 months ago

What?

Add functionality to upload json data as zipped files to Neptune.

Why?

This allows for storing json data from many distributed experiments runs in a central place. GIven a Neptune tag the json files can then be downloaded, joined, aggregated and plotted with tooling from MARL-eval

How?

Add a zip_and_upload_json method to the Neptune logger.

Extra

Simplified the path where json files are stored locally.