From 5bbdf02750b5e904068ee162e5ef4c8879a15fd4 Mon Sep 17 00:00:00 2001
From: lbenhorin <lbenhorin@nvidia.com>
Date: Fri, 3 Jul 2026 15:31:18 +0300
Subject: [PATCH] docs(groot): update training command with  image
 transformation parameters

---
 docs/source/groot.mdx | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/docs/source/groot.mdx b/docs/source/groot.mdx
index 25fe3e7fd..831f42d08 100644
--- a/docs/source/groot.mdx
+++ b/docs/source/groot.mdx
@@ -116,11 +116,21 @@ GR00T N1.7 has demonstrated strong performance on the LIBERO benchmark suite. To
 Example training command for a LIBERO suite (here `libero_spatial`):
 
 ```bash
+IMAGE_TRANSFORMS='{
+  "brightness": {"weight": 1.0, "type": "ColorJitter", "kwargs": {"brightness": [0.7, 1.3]}},
+  "contrast":   {"weight": 1.0, "type": "ColorJitter", "kwargs": {"contrast":   [0.6, 1.4]}},
+  "saturation": {"weight": 1.0, "type": "ColorJitter", "kwargs": {"saturation": [0.5, 1.5]}},
+  "hue":        {"weight": 1.0, "type": "ColorJitter", "kwargs": {"hue":        [-0.08, 0.08]}}
+}'
+
 lerobot-train \
   --dataset.repo_id=IPEC-COMMUNITY/libero_spatial_no_noops_1.0.0_lerobot \
   --dataset.root=/datasets/libero_spatial \
   --dataset.revision=main \
   --dataset.video_backend=pyav \
+  --dataset.image_transforms.enable=true \
+  --dataset.image_transforms.max_num_transforms=4 \
+  --dataset.image_transforms.tfs="$IMAGE_TRANSFORMS" \
   --policy.type=groot \
   --policy.base_model_path=nvidia/GR00T-N1.7-3B \
   --policy.embodiment_tag=libero_sim \
@@ -140,10 +150,7 @@ lerobot-train \
   --prefetch_factor=2 \
   --persistent_workers=true \
   --output_dir=$OUTPUT_DIR \
-  --job_name=$JOB_NAME \
-  --dataset.image_transforms.enable=true \
-  --dataset.image_transforms.max_num_transforms=4 \
-  --dataset.image_transforms.tfs='{"brightness":{"weight":1.0,"type":"ColorJitter","kwargs":{"brightness":[0.7,1.3]}},"contrast":{"weight":1.0,"type":"ColorJitter","kwargs":{"contrast":[0.6,1.4]}},"saturation":{"weight":1.0,"type":"ColorJitter","kwargs":{"saturation":[0.5,1.5]}},"hue":{"weight":1.0,"type":"ColorJitter","kwargs":{"hue":[-0.08,0.08]}}}'
+  --job_name=$JOB_NAME
 ```
 
 ### GR00T N1.7 LIBERO Results