search

google / evojax

Apache License 2.0
846 stars 87 forks source link

Native implementation in JAX of Augmented Random Search #47

Closed EdoardoPona closed 2 years ago

EdoardoPona commented 2 years ago

Test results

Note for MNIST I halved the batch size and doubled the iterations due to memory issues. Benchmark Params Results (avg.)
CartPole (easy) 900 (max_iter=1000) Link 910
CartPole (hard) 600 (max_iter=2000) Link 558.02
MNIST 0.90 (max_iter=4000) Link 0.92
Brax Ant 3000 (max_iter=700) Link 4129.83
Waterworld 6 (max_iter=2000) Link 7.29
Waterworld (MA) 2 (max_iter=2000) Link 1.68
lerrytang commented 2 years ago

Hi, thank you for sending the PR. We will proceed to review after you fix the lint/test errors.

alantian commented 2 years ago

Hey @EdoardoPona thanks for your contribution!

Would you mind also including the the content of the test result table you show here to scripts/benchmarks/Readme.md in PR (as modified files)?

EdoardoPona commented 2 years ago

Done!

lerrytang commented 2 years ago

@EdoardoPona Thanks for your contribution! Would you like to send a pr to update this page?