PKU-Alignment / omnisafe

OmniSafe is an infrastructural framework for accelerating SafeRL research.
https://www.omnisafe.ai
Apache License 2.0
883 stars 129 forks source link

[Question] A question about epoch visualization during the training process #339

Open CrazyTrumpet10 opened 1 month ago

CrazyTrumpet10 commented 1 month ago

Required prerequisites

Questions

epoch_1 1716970569230 Hello, my research direction is path planning, I would like to ask how to render the visualization of each epoch trajectory image, check the description of the document does not seem to find the relevant instructions, look forward to your reply, thank you very much! Every time I render the graph, it is the initialized environment with no trajectories, I also try to get the coordinates of the intelligences for each epoch, and I can get them, but I can't render them in the environment graph, and I would like to ask how to render the trajectories properly?

Gaiejj commented 1 month ago

I've grasped your requirement, but currently, OmniSafe doesn't offer an interface for visualizing during the training process or saving rendered videos after each epoch. We acknowledge this as a valuable feature and intend to support it in future updates.

In the meantime, to address your current needs, you might consider saving the model weights at the end of each epoch. By setting save_model_freq in the algorithm yaml file.

Post-training, you can individually load these saved weights and proceed with visualizations. This approach allows you to retrospectively analyze the progress and performance of the model at every stage.