[Question] How can I record data at each episode during the training ?

PKU-Alignment / omnisafe

OmniSafe is an infrastructural framework for accelerating SafeRL research.

https://www.omnisafe.ai

Apache License 2.0

903 stars 130 forks source link

[Question] How can I record data at each episode during the training ? #349

Open Aso2510 opened 2 weeks ago

Aso2510 commented 2 weeks ago

Required prerequisites

[X] I have read the documentation https://omnisafe.readthedocs.io.
[X] I have searched the Issue Tracker and Discussions that this hasn't already been reported. (+1 or comment there if it has.)
[X] Consider asking first in a Discussion.

Questions

I tried to record training data ( RE, Cost) as per episode not per epoch as provided in the progress.csv.

Is this supported? or Is there anyway to do that?

Any help is really appreciated

Gaiejj commented 2 weeks ago

This is a very nice suggestion, and we will include this feature in future versions. For now, you may consider adding env_spec_log to the SafetyGymnasiumEnv class located in omnisafe/envs/safety_gymnasium_env.py to log the environment specifications. For more details, please refer to our tutorial 😊