FLAIROx / JaxMARL

Multi-Agent Reinforcement Learning with JAX
Apache License 2.0
393 stars 68 forks source link

How to integrate marl-eval in JaxMARL #59

Closed fmxFranky closed 6 months ago

fmxFranky commented 6 months ago

Hi! Amazing work you've done here! I wanna integrate marl-eval in JaxMARL, can you give me some snippets about it? Thank you very much!

amacrutherford commented 6 months ago

Hey! So as far as I am aware marl-eval needs the experimental results data to be in a particular form which could be achieved by amending our logging code to fit their requirements. It is on our list to do but is not a priority for now. Let me know if this helps!

RuanJohn commented 6 months ago

Hey there @fmxFranky, we just made a new MARL-eval release today that now adds a logger that will handle most of the heavy lifting for you in terms of logging in our required format. Please have a look at the logger and some other new JSON tools we have added to repo here. For an example of how we set up the MARL-eval JSON logger in Mava please also see here.