policy callback 'policy_params_fn' for other algorithms?

google / brax

Massively parallel rigidbody physics simulation on accelerator hardware.

Apache License 2.0

2.14k stars 234 forks source link

Open Rian-Jo opened 1 month ago

Rian-Jo commented 1 month ago

currently, policy_params_fn exists only in ppo algorithm. are you going to update the policy clalback function to other algorithms too?

btaba commented 1 month ago

Hi @Rian-Jo , wasn't planning on updating that any time soon, feel free to open up a PR!