From d44f3a3bd93b5e8521a3104ed6b31b16a4755abd Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Mon, 19 Jan 2026 15:48:14 +0000 Subject: [PATCH] update --- .../pi05_full/annotate/annotate_libero.sh | 27 +++++++++---------- 1 file changed, 13 insertions(+), 14 deletions(-) diff --git a/src/lerobot/policies/pi05_full/annotate/annotate_libero.sh b/src/lerobot/policies/pi05_full/annotate/annotate_libero.sh index 557cacb1e..ab3bad68c 100644 --- a/src/lerobot/policies/pi05_full/annotate/annotate_libero.sh +++ b/src/lerobot/policies/pi05_full/annotate/annotate_libero.sh @@ -16,12 +16,12 @@ TEMPERATURE=0.9 SAMPLE_INTERVAL=5.0 # generate dialogue every 1 second (all episodes processed) # Run subtask annotation -python /admin/home/jade_choghari/lerobot/src/lerobot/policies/pi05_full/annotate/subtask_annotate.py \ - --repo-id "$REPO_ID" \ - --video-key observation.images.image \ - --output-dir "$OUTPUT_DIR" \ - --output-repo-id "jadechoghari/libero-annotate" \ - --batch-size "$BATCH_SIZE" \ +# python /admin/home/jade_choghari/lerobot/src/lerobot/policies/pi05_full/annotate/subtask_annotate.py \ +# --repo-id "$REPO_ID" \ +# --video-key observation.images.image \ +# --output-dir "$OUTPUT_DIR" \ +# --output-repo-id "jadechoghari/libero-annotate" \ +# --batch-size "$BATCH_SIZE" \ # run synthetic data generation (all episodes processed) # python examples/dataset/annotate_pgen.py \ # --repo-id "$REPO_ID" \ @@ -40,11 +40,10 @@ python /admin/home/jade_choghari/lerobot/src/lerobot/policies/pi05_full/annotate # add --push-to-hub flag # efficient batch processing: 4 episodes at once -# python examples/dataset/annotate_pgen.py \ -# --repo-id "$REPO_ID" \ -# --model "$MODEL" \ -# --output-dir "$OUTPUT_DIR" \ -# --video-mode \ -# --video-key observation.images.up \ -# --video-batch-size "$BATCH_SIZE" \ -# --sample-interval 1.0 +python /admin/home/jade_choghari/lerobot/src/lerobot/policies/pi05_full/annotate/high_level_annotate.py \ + --repo-id "$REPO_ID" \ + --output-dir "$OUTPUT_DIR" \ + --video-mode \ + --video-key observation.images.image \ + --video-batch-size "$BATCH_SIZE" \ + --sample-interval 5.0