How to visualize when training like isaacGYM

[kassasin]

I want to visualize training process (not only rewards but also robot moving), does brax suport this>

[btaba]

Hi @kassasin , you can use the policy_params_fn callback to render videos for every eval that gets run

[kassasin]

Thanks，I will take a try .

[kassasin]

I think there have some differences . I can use "policy_params_fn" to save policy and use it to visualize some robot, but I think it's not necessary. We can use every enviroment's robot state to visualize .Is my understanding correct? @btaba