From 7ff93e8a51e90dfdef69fdbc703e40d4a67bb789 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Sat, 8 Jun 2024 12:17:28 +0200 Subject: [PATCH] Update examples/real_robot_example/0_record_training_data.py Co-authored-by: Remi --- examples/real_robot_example/0_record_training_data.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/real_robot_example/0_record_training_data.py b/examples/real_robot_example/0_record_training_data.py index 8c392941..6f483e4b 100644 --- a/examples/real_robot_example/0_record_training_data.py +++ b/examples/real_robot_example/0_record_training_data.py @@ -39,7 +39,9 @@ revision = args.revision out_data = DATA_DIR / repo_id +# During data collection, frames are stored as png images in `images_dir` images_dir = out_data / "images" +# After data collection, png images of each episode are encoded into a mp4 file stored in `videos_dir` videos_dir = out_data / "videos" meta_data_dir = out_data / "meta_data"