Skip to content

Conversation

@heslami
Copy link
Contributor

@heslami heslami commented Dec 17, 2025

Changes proposed in this pull request:

  • Adds the benchmark measurements for Orin.

Commands used to run the benchmark script on Orin:

python scripts/deployment/export_onnx_n1d6.py --model_path nvidia/GR00T-N1.6-3B --dataset_path demo_data/gr1.PickNPlace --output_dir ./groot_n1d6_onnx --embodiment_tag gr1

python scripts/deployment/build_tensorrt_engine.py --onnx ./groot_n1d6_onnx/dit_model.onnx --engine ./groot_n1d6_onnx/dit_model_bf16.trt --precision bf16

TORCHINDUCTOR_COMPILE_THREADS=1 python scripts/deployment/benchmark_inference.py --model_path nvidia/GR00T-N1.6-3B --dataset_path demo_data/gr1.PickNPlace --trt_engine_path ./groot_n1d6_onnx/dit_model_bf16.trt 
  • I've read and followed all steps in the Making a pull request
    section of the CONTRIBUTING docs.
  • I've updated or added any relevant docstrings.
  • If this PR fixes a bug, I've added a test that will fail without my fix.
  • If this PR adds a new feature, I've added tests that sufficiently cover my new functionality.

Copy link
Member

@youliangtan youliangtan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@youliangtan youliangtan merged commit d483f00 into NVIDIA:main Dec 18, 2025
1 check passed
vertix pushed a commit to Positronic-Robotics/gr00t that referenced this pull request Jan 2, 2026
vertix pushed a commit to Positronic-Robotics/gr00t that referenced this pull request Jan 2, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants