search

google / evojax

Apache License 2.0
834 stars 85 forks source link

Advanced loggers #11

Closed danielgafni closed 2 years ago

danielgafni commented 2 years ago

It would be nice to use Tensorboard or any other advanced logging tool with evojax. Looks like it should be straightforward to allow the user to implement a log_reward function that could be passed to the trainer.

I'm willing to implement this in a PR.

lerrytang commented 2 years ago

Hi, thanks for the suggestion, it sounds great and I'm looking forward to your PR. One principle to keep in mind is that we try to make EvoJAX have few dependencies.