leggedrobotics / legged_gym

Isaac Gym Environments for Legged Robots
Other
1.26k stars 363 forks source link

Capturing video during training #66

Closed dmtrung14 closed 1 week ago

dmtrung14 commented 4 months ago

I am following the instructions here: https://github.com/nvidia-omniverse/IsaacGymEnvs?tab=readme-ov-file#capture-videos to capture video during training, by adding:

envs.is_vector_env = True
envs = gym.wrappers.RecordVideo(
    envs,
    "./videos",
    step_trigger=lambda step: step % 10000 == 0, # record the videos every 10000 steps
    video_length=100  # for each video record up to 100 steps
)

to task_registry.make_envs but it only creates json files with empty videos.

Any help is appreciated!

dmtrung14 commented 4 months ago

So one key notice I realized is that the ends here are wrapped in VecEnv class, so that example is probably not working. However, I still want to capture videos during training/playback since I'm working on a remote desktop.