Missing code for extracting position and velocity using OD module #4
Description
This is a wonderful code repository with many impressive capabilities. I sincerely appreciate you sharing it.
When testing the code, I noticed the section for utilizing the OD module to extract velocity and position in[save_offline_dataset.py]&[symbolic_finetuning_guided.py] does not seem to be included, so do as the "save teacher dataset" code. If it's not too much trouble, adding that code would significantly improve the functionality. Please let me know if you would be willing to add it at your convenience.
Thank you again for providing this great resource. Your work is incredibly valuable to the community. I look forward to seeing future updates!