opentau 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- opentau/__init__.py +179 -0
- opentau/__version__.py +24 -0
- opentau/configs/__init__.py +19 -0
- opentau/configs/default.py +297 -0
- opentau/configs/libero.py +113 -0
- opentau/configs/parser.py +393 -0
- opentau/configs/policies.py +297 -0
- opentau/configs/reward.py +42 -0
- opentau/configs/train.py +370 -0
- opentau/configs/types.py +76 -0
- opentau/constants.py +52 -0
- opentau/datasets/__init__.py +84 -0
- opentau/datasets/backward_compatibility.py +78 -0
- opentau/datasets/compute_stats.py +333 -0
- opentau/datasets/dataset_mixture.py +460 -0
- opentau/datasets/factory.py +232 -0
- opentau/datasets/grounding/__init__.py +67 -0
- opentau/datasets/grounding/base.py +154 -0
- opentau/datasets/grounding/clevr.py +110 -0
- opentau/datasets/grounding/cocoqa.py +130 -0
- opentau/datasets/grounding/dummy.py +101 -0
- opentau/datasets/grounding/pixmo.py +177 -0
- opentau/datasets/grounding/vsr.py +141 -0
- opentau/datasets/image_writer.py +304 -0
- opentau/datasets/lerobot_dataset.py +1910 -0
- opentau/datasets/online_buffer.py +442 -0
- opentau/datasets/push_dataset_to_hub/utils.py +132 -0
- opentau/datasets/sampler.py +99 -0
- opentau/datasets/standard_data_format_mapping.py +278 -0
- opentau/datasets/transforms.py +330 -0
- opentau/datasets/utils.py +1243 -0
- opentau/datasets/v2/batch_convert_dataset_v1_to_v2.py +887 -0
- opentau/datasets/v2/convert_dataset_v1_to_v2.py +829 -0
- opentau/datasets/v21/_remove_language_instruction.py +109 -0
- opentau/datasets/v21/batch_convert_dataset_v20_to_v21.py +60 -0
- opentau/datasets/v21/convert_dataset_v20_to_v21.py +183 -0
- opentau/datasets/v21/convert_stats.py +150 -0
- opentau/datasets/video_utils.py +597 -0
- opentau/envs/__init__.py +18 -0
- opentau/envs/configs.py +178 -0
- opentau/envs/factory.py +99 -0
- opentau/envs/libero.py +439 -0
- opentau/envs/utils.py +204 -0
- opentau/optim/__init__.py +16 -0
- opentau/optim/factory.py +43 -0
- opentau/optim/optimizers.py +121 -0
- opentau/optim/schedulers.py +140 -0
- opentau/planner/__init__.py +82 -0
- opentau/planner/high_level_planner.py +366 -0
- opentau/planner/utils/memory.py +64 -0
- opentau/planner/utils/utils.py +65 -0
- opentau/policies/__init__.py +24 -0
- opentau/policies/factory.py +172 -0
- opentau/policies/normalize.py +315 -0
- opentau/policies/pi0/__init__.py +19 -0
- opentau/policies/pi0/configuration_pi0.py +250 -0
- opentau/policies/pi0/modeling_pi0.py +994 -0
- opentau/policies/pi0/paligemma_with_expert.py +516 -0
- opentau/policies/pi05/__init__.py +20 -0
- opentau/policies/pi05/configuration_pi05.py +231 -0
- opentau/policies/pi05/modeling_pi05.py +1257 -0
- opentau/policies/pi05/paligemma_with_expert.py +572 -0
- opentau/policies/pretrained.py +315 -0
- opentau/policies/utils.py +123 -0
- opentau/policies/value/__init__.py +18 -0
- opentau/policies/value/configuration_value.py +170 -0
- opentau/policies/value/modeling_value.py +512 -0
- opentau/policies/value/reward.py +87 -0
- opentau/policies/value/siglip_gemma.py +221 -0
- opentau/scripts/actions_mse_loss.py +89 -0
- opentau/scripts/bin_to_safetensors.py +116 -0
- opentau/scripts/compute_max_token_length.py +111 -0
- opentau/scripts/display_sys_info.py +90 -0
- opentau/scripts/download_libero_benchmarks.py +54 -0
- opentau/scripts/eval.py +877 -0
- opentau/scripts/export_to_onnx.py +180 -0
- opentau/scripts/fake_tensor_training.py +87 -0
- opentau/scripts/get_advantage_and_percentiles.py +220 -0
- opentau/scripts/high_level_planner_inference.py +114 -0
- opentau/scripts/inference.py +70 -0
- opentau/scripts/launch_train.py +63 -0
- opentau/scripts/libero_simulation_parallel.py +356 -0
- opentau/scripts/libero_simulation_sequential.py +122 -0
- opentau/scripts/nav_high_level_planner_inference.py +61 -0
- opentau/scripts/train.py +379 -0
- opentau/scripts/visualize_dataset.py +294 -0
- opentau/scripts/visualize_dataset_html.py +507 -0
- opentau/scripts/zero_to_fp32.py +760 -0
- opentau/utils/__init__.py +20 -0
- opentau/utils/accelerate_utils.py +79 -0
- opentau/utils/benchmark.py +98 -0
- opentau/utils/fake_tensor.py +81 -0
- opentau/utils/hub.py +209 -0
- opentau/utils/import_utils.py +79 -0
- opentau/utils/io_utils.py +137 -0
- opentau/utils/libero.py +214 -0
- opentau/utils/libero_dataset_recorder.py +460 -0
- opentau/utils/logging_utils.py +180 -0
- opentau/utils/monkey_patch.py +278 -0
- opentau/utils/random_utils.py +244 -0
- opentau/utils/train_utils.py +198 -0
- opentau/utils/utils.py +471 -0
- opentau-0.1.0.dist-info/METADATA +161 -0
- opentau-0.1.0.dist-info/RECORD +108 -0
- opentau-0.1.0.dist-info/WHEEL +5 -0
- opentau-0.1.0.dist-info/entry_points.txt +2 -0
- opentau-0.1.0.dist-info/licenses/LICENSE +508 -0
- opentau-0.1.0.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,161 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: opentau
|
|
3
|
+
Version: 0.1.0
|
|
4
|
+
Summary: OpenTau: Tensor's VLA Training Infrastructure for Real-World Robotics in Pytorch
|
|
5
|
+
Author-email: Shuheng Liu <wish1104@icloud.com>, William Yue <williamyue37@gmail.com>, Akshay Shah <akshayhitendrashah@gmail.com>, Xingrui Gu <xingrui_gu@berkeley.edu>
|
|
6
|
+
License: Apache-2.0
|
|
7
|
+
Project-URL: homepage, https://github.com/TensorAuto/OpenTau
|
|
8
|
+
Project-URL: issues, https://github.com/TensorAuto/OpenTau/issues
|
|
9
|
+
Project-URL: huggingface, https://huggingface.co/TensorAuto
|
|
10
|
+
Keywords: robotics,deep learning,pytorch
|
|
11
|
+
Classifier: Development Status :: 3 - Alpha
|
|
12
|
+
Classifier: Intended Audience :: Developers
|
|
13
|
+
Classifier: Intended Audience :: Education
|
|
14
|
+
Classifier: Intended Audience :: Science/Research
|
|
15
|
+
Classifier: Topic :: Software Development :: Build Tools
|
|
16
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
17
|
+
Classifier: License :: OSI Approved :: Apache Software License
|
|
18
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
19
|
+
Requires-Python: ==3.10.*
|
|
20
|
+
Description-Content-Type: text/markdown
|
|
21
|
+
License-File: LICENSE
|
|
22
|
+
Requires-Dist: cmake>=3.29.0.1
|
|
23
|
+
Requires-Dist: datasets>=2.19.0
|
|
24
|
+
Requires-Dist: deepdiff>=7.0.1
|
|
25
|
+
Requires-Dist: diffusers>=0.27.2
|
|
26
|
+
Requires-Dist: draccus>=0.10.0
|
|
27
|
+
Requires-Dist: einops>=0.8.0
|
|
28
|
+
Requires-Dist: flask>=3.0.3
|
|
29
|
+
Requires-Dist: gdown>=5.1.0
|
|
30
|
+
Requires-Dist: h5py>=3.10.0
|
|
31
|
+
Requires-Dist: huggingface-hub[cli,hf-transfer]>=0.27.1; python_version < "4.0"
|
|
32
|
+
Requires-Dist: imageio[ffmpeg]>=2.34.0
|
|
33
|
+
Requires-Dist: jsonlines>=4.0.0
|
|
34
|
+
Requires-Dist: numba>=0.62.0
|
|
35
|
+
Requires-Dist: omegaconf>=2.3.0
|
|
36
|
+
Requires-Dist: opencv-python-headless>=4.9.0
|
|
37
|
+
Requires-Dist: packaging>=24.2
|
|
38
|
+
Requires-Dist: av>=12.0.5
|
|
39
|
+
Requires-Dist: pymunk>=6.6.0
|
|
40
|
+
Requires-Dist: pynput>=1.7.7
|
|
41
|
+
Requires-Dist: pyzmq>=26.2.1
|
|
42
|
+
Requires-Dist: rerun-sdk>=0.21.0
|
|
43
|
+
Requires-Dist: termcolor>=2.4.0
|
|
44
|
+
Requires-Dist: torch<2.8.0,>=2.7.1
|
|
45
|
+
Requires-Dist: torchcodec<0.5.0,>=0.4.0; sys_platform != "win32" and (sys_platform != "linux" or (platform_machine != "aarch64" and platform_machine != "arm64" and platform_machine != "armv7l")) and (sys_platform != "darwin" or platform_machine != "x86_64")
|
|
46
|
+
Requires-Dist: torchvision<0.23.0,>=0.22.1
|
|
47
|
+
Requires-Dist: wandb>=0.16.3
|
|
48
|
+
Requires-Dist: zarr>=2.17.0
|
|
49
|
+
Requires-Dist: scikit-learn>=1.7.1
|
|
50
|
+
Requires-Dist: onnx>=1.18.0
|
|
51
|
+
Requires-Dist: onnxruntime>=1.22.1; sys_platform == "darwin" or platform_machine == "arm64" or platform_machine == "aarch64"
|
|
52
|
+
Requires-Dist: onnxruntime-gpu>=1.22.0; (sys_platform == "linux" and platform_machine == "x86_64") or (sys_platform == "win32" and (platform_machine == "AMD64" or platform_machine == "x86_64"))
|
|
53
|
+
Requires-Dist: onnxscript>=0.3.1
|
|
54
|
+
Requires-Dist: onnx-ir>=0.1.4
|
|
55
|
+
Requires-Dist: opentau-transformers==4.53.3
|
|
56
|
+
Requires-Dist: scipy>=1.15.2
|
|
57
|
+
Requires-Dist: pytest>=8.1.0
|
|
58
|
+
Requires-Dist: pytest-cov>=5.0.0
|
|
59
|
+
Requires-Dist: pyserial>=3.5
|
|
60
|
+
Requires-Dist: pytest-xdist>=3.8.0
|
|
61
|
+
Requires-Dist: scikit-image>=0.23.2
|
|
62
|
+
Requires-Dist: pandas>=2.2.2
|
|
63
|
+
Requires-Dist: accelerate>=1.4.0
|
|
64
|
+
Requires-Dist: deepspeed>=0.17.1
|
|
65
|
+
Provides-Extra: dev
|
|
66
|
+
Requires-Dist: pre-commit>=3.7.0; extra == "dev"
|
|
67
|
+
Requires-Dist: debugpy>=1.8.1; extra == "dev"
|
|
68
|
+
Requires-Dist: pytest>=8.1.0; extra == "dev"
|
|
69
|
+
Requires-Dist: pytest-cov>=5.0.0; extra == "dev"
|
|
70
|
+
Requires-Dist: pyserial>=3.5; extra == "dev"
|
|
71
|
+
Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
|
|
72
|
+
Requires-Dist: sphinx>=8.1.3; extra == "dev"
|
|
73
|
+
Requires-Dist: sphinx-rtd-theme>=3.0.1; extra == "dev"
|
|
74
|
+
Requires-Dist: sphinx-copybutton>=0.5.2; extra == "dev"
|
|
75
|
+
Requires-Dist: sphinx-design>=0.6.1; extra == "dev"
|
|
76
|
+
Requires-Dist: myst-parser>=4.0.0; extra == "dev"
|
|
77
|
+
Provides-Extra: openai
|
|
78
|
+
Requires-Dist: openai>=1.0.0; extra == "openai"
|
|
79
|
+
Requires-Dist: python-dotenv>=1.0.0; extra == "openai"
|
|
80
|
+
Provides-Extra: libero
|
|
81
|
+
Requires-Dist: cmake<4; extra == "libero"
|
|
82
|
+
Requires-Dist: ninja; extra == "libero"
|
|
83
|
+
Requires-Dist: bddl==1.0.1; extra == "libero"
|
|
84
|
+
Requires-Dist: easydict==1.9; extra == "libero"
|
|
85
|
+
Requires-Dist: future==0.18.2; extra == "libero"
|
|
86
|
+
Requires-Dist: matplotlib>=3.5.3; extra == "libero"
|
|
87
|
+
Requires-Dist: robomimic==0.2.0; extra == "libero"
|
|
88
|
+
Requires-Dist: robosuite==1.4.0; extra == "libero"
|
|
89
|
+
Requires-Dist: thop==0.1.1.post2209072238; extra == "libero"
|
|
90
|
+
Requires-Dist: mujoco>=3.3.5; extra == "libero"
|
|
91
|
+
Requires-Dist: PyOpenGL==3.1.7; extra == "libero"
|
|
92
|
+
Requires-Dist: libero; extra == "libero"
|
|
93
|
+
Requires-Dist: numpy<2; extra == "libero"
|
|
94
|
+
Requires-Dist: gym<0.27,>=0.25; extra == "libero"
|
|
95
|
+
Requires-Dist: pyopengl-accelerate==3.1.7; sys_platform == "linux" and extra == "libero"
|
|
96
|
+
Requires-Dist: gymnasium[other]>=0.29; extra == "libero"
|
|
97
|
+
Dynamic: license-file
|
|
98
|
+
|
|
99
|
+
<p align="center">
|
|
100
|
+
<a href="https://www.tensor.auto">
|
|
101
|
+
<img src="assets/logo.png" alt="Logo">
|
|
102
|
+
</a>
|
|
103
|
+
</p>
|
|
104
|
+
|
|
105
|
+
# OpenTau - Train VLA models with state-of-the-art techniques by Tensor
|
|
106
|
+
|
|
107
|
+
At Tensor, we are pushing the frontier of large foundation models for physical AI. In robot learning, a vision-language-action (VLA) model is a multimodal foundation model that integrates vision, language, and action. Today, VLA represents the leading approach for embodied AI, spanning autonomous driving, robot manipulation, and navigation.
|
|
108
|
+
|
|
109
|
+
OpenTau is Tensor’s open-source training toolchain for frontier VLA models—designed to make training reproducible, accessible, and scalable. At Tensor, we believe in open research and reproducible progress for the robotics community. By open-sourcing our training toolchain, we aim to expand knowledge sharing and accelerate scientific progress that others can reproduce.
|
|
110
|
+
|
|
111
|
+
Whether you use the official OpenPi codebase or LeRobot’s reimplementation, you may still be missing key components. OpenTau implements these key capabilities in one place:
|
|
112
|
+
|
|
113
|
+
- Co-training on an adjustable mixture of heterogeneous datasets
|
|
114
|
+
- Discrete actions for fast VLM convergence in $\pi_{0.5}$
|
|
115
|
+
- Knowledge insulation between the VLM backbone and the action expert
|
|
116
|
+
- Dropout in the VLM to reduce overfitting
|
|
117
|
+
- A reinforcement learning pipeline described in $\pi^*_{0.6}$
|
|
118
|
+
- And more...
|
|
119
|
+
|
|
120
|
+
OpenTau ($\tau$) is a tool developed by *[Tensor][1]* to bridge this gap, and we also use it internally to train our proprietary in-house models. Our goal is to help you train VLAs on any dataset while fully leveraging state-of-the-art techniques. We plan to continuously upgrade this repository to keep pace with the state of the art in the robotics community.
|
|
121
|
+
|
|
122
|
+
| Features | OpenPi | LeRobot | **OpenTau** |
|
|
123
|
+
| -------------------------------------------------------: | :---------------------: | :------------------------------: | :---------: |
|
|
124
|
+
| Co-training with Heterogeneous Datasets | ❌ | ❌ | ✅ |
|
|
125
|
+
| Discrete Actions Training in $\pi_{0.5}$ | ❌ | ❌ | ✅ |
|
|
126
|
+
| Knowledge Insulation (KI) between VLM and Action Decoder | ❌ | ❌ | ✅ |
|
|
127
|
+
| Dropout Layers in PaliGemma | ✅ (Jax) <br>❌ (PyTorch) | ❌ | ✅ |
|
|
128
|
+
| Multi-Node and Multi-GPU Training | ❌ | ✅ | ✅ |
|
|
129
|
+
| Fully Functioning $\pi_{0.5}$ Checkpoint | ✅ | ❌ <br> (Missing Text Embeddings) | ✅ |
|
|
130
|
+
| Simulation Environments for Evaluating Models | ❌ | ✅ | ✅ |
|
|
131
|
+
| $\pi^{*}_{0.6}$ style Reinforcement Learning Pipeline | ❌ | ❌ | ✅ |
|
|
132
|
+
| Framework | Jax / PyTorch | PyTorch | PyTorch |
|
|
133
|
+
|
|
134
|
+
## Quick Start
|
|
135
|
+
If you are familiar with LeRobot, getting started with OpenTau is very easy.
|
|
136
|
+
Because OpenTau is a fork of the popular LeRobot repository, any LeRobot-compliant policy and dataset can be used directly with OpenTau.
|
|
137
|
+
Check out our documentation to get started quickly.
|
|
138
|
+
We provide a quick start guide to help you get started with OpenTau.
|
|
139
|
+
|
|
140
|
+
For using local notebooks to train and evaluate models, find the notebooks at `notebooks/pi05_training.ipynb` and `notebooks/pi05_evaluation_only.ipynb`.
|
|
141
|
+
|
|
142
|
+
For using the Google Colab notebooks to train and evaluate models, find the colab notebooks here: [pi05_training](https://colab.research.google.com/drive/1DeU0lNnEzs1KHo0Nkgh4YKBr-xu9moBM?usp=sharing) and [pi05_evaluation_only](https://colab.research.google.com/drive/1U_AyuH9WYMT4anEWvsOtIT7g01jA0WGm?usp=sharing) respectively.
|
|
143
|
+
|
|
144
|
+
## Checkpoints
|
|
145
|
+
We provide fully functioning $\pi_{0.5}$ checkpoints trained with high success rates. We plan to release more models in the near future.
|
|
146
|
+
|
|
147
|
+
| Model Checkpoint | Description | Success Rate (%) |
|
|
148
|
+
|-------------------------------|---------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------|
|
|
149
|
+
| [TensorAuto/tPi0.5-libero][2] | A $\pi_{0.5}$ model checkpoint trained on the LIBERO dataset with discrete actions and knowledge insulation. | 98.4% (10) <br> 97.6% (Goal) <br> 100% (Object) <br> 98% (Spatial) |
|
|
150
|
+
| [TensorAuto/pi05_base][5] | A $\pi_{0.5}$ model checkpoint converted from the official openpi checkpoint, with language embeddings added. | N/A |
|
|
151
|
+
| More coming soon... | | |
|
|
152
|
+
|
|
153
|
+
## Acknowledgements
|
|
154
|
+
|
|
155
|
+
This project builds on the $\pi$ series of [papers][3] and many other open-source efforts—especially [LeRobot][4]—for re-implementing the $\pi$ models and helping standardize training infrastructure. OpenTau extends these foundations to provide a more accessible, comprehensive toolchain for training vision-language-action agents.
|
|
156
|
+
|
|
157
|
+
[1]: https://www.tensor.ai
|
|
158
|
+
[2]: https://huggingface.co/TensorAuto/tPi0.5-libero
|
|
159
|
+
[3]: https://www.pi.website/blog
|
|
160
|
+
[4]: https://huggingface.co/lerobot
|
|
161
|
+
[5]: https://huggingface.co/TensorAuto/pi05_base
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
opentau/__init__.py,sha256=KktmFuQQqDhhzcJKczJ99crma9qjxiJZUPv-_d0ZeLU,6697
|
|
2
|
+
opentau/__version__.py,sha256=junxoss59Jz_hmg3YnzhpVk_Q5Fo6uha23P1ET81N1c,889
|
|
3
|
+
opentau/constants.py,sha256=-_CbJujCp6hbBjJHgYMguCTcSAkVkmdpM4wHqZp7vRQ,2020
|
|
4
|
+
opentau/configs/__init__.py,sha256=hC-KkeCfq1mtMw9WjPCZfOTxrzQWW7hAa1w8BRC_Bqw,784
|
|
5
|
+
opentau/configs/default.py,sha256=MEoZyzK8olVXVHDy3FHQKK43-ULq2UQv3almymbmdCI,14387
|
|
6
|
+
opentau/configs/libero.py,sha256=CrRfiCBYOw7hVqv6orH_ahNyQudj6iyqHtZM9YpdvzE,4688
|
|
7
|
+
opentau/configs/parser.py,sha256=Pb7sw6yx38F31Aqw1J7wK6BRzfBA7DutywShyP_t9bY,14890
|
|
8
|
+
opentau/configs/policies.py,sha256=06oUJx0B4V6krRwyjH1goTYM3RIpRozg4SSwcVJurG4,11667
|
|
9
|
+
opentau/configs/reward.py,sha256=t7S8_RpEy31fAP4v_ygB-ETvaUR6OyrXmS1JNSz3cOk,1537
|
|
10
|
+
opentau/configs/train.py,sha256=a-c-s2zpCUkK8n0DqBPGGaJ87UcVNF02RcLa1pH843Y,18049
|
|
11
|
+
opentau/configs/types.py,sha256=DvKasR2v0ecSmozL0YD4S-64OeuDYhVBhtspxUDV5u0,2453
|
|
12
|
+
opentau/datasets/__init__.py,sha256=oLfV9vfFOg7o2XIRFiN5zOf529FafIkPwqFG7iUX4gc,4248
|
|
13
|
+
opentau/datasets/backward_compatibility.py,sha256=ENVQk9QDPCip6NfAxNF6Vo4VvyCWb0acV-ZxcJBsB6o,3459
|
|
14
|
+
opentau/datasets/compute_stats.py,sha256=N359TDuJicLKMtxxy0JVEcUtnTOB57gL5G8e9Dq0gMQ,13069
|
|
15
|
+
opentau/datasets/dataset_mixture.py,sha256=8UWjY9oKn9jEMe-e9Dy6no1p_21H0kXKv8A10Ku_8_o,19850
|
|
16
|
+
opentau/datasets/factory.py,sha256=NKWpbuNBve0PsmK1midj8g1IpQapeHn-VrxCOC3X4eI,10480
|
|
17
|
+
opentau/datasets/image_writer.py,sha256=JYCkImHFYpLuE88t16cYqXqQS7EHS7g6kLWXPCJmWgw,11072
|
|
18
|
+
opentau/datasets/lerobot_dataset.py,sha256=rz_3BcXqpcIzYr0NEVmkfLf1dY7vcTdo6zuV1CZkIuI,84747
|
|
19
|
+
opentau/datasets/online_buffer.py,sha256=x14P8tBz25s-hRlE8loFJs5CAvh65RGWeogF271hiF0,19671
|
|
20
|
+
opentau/datasets/sampler.py,sha256=5g-6prsWItVjqkt1J7mA9JPNQPhDSFx3r6rA4JphP9U,4012
|
|
21
|
+
opentau/datasets/standard_data_format_mapping.py,sha256=wEKilksMUjJGeIhvyLuR9qhyhtiJMK1e1AzCkbyx-l4,9667
|
|
22
|
+
opentau/datasets/transforms.py,sha256=pr_8vOEDUoWu7aOUdnI0_wgetsFuie3I2UYFrcStG1k,12976
|
|
23
|
+
opentau/datasets/utils.py,sha256=bZ0Q8KPZMWe9fLdrqJqslgDxI9sa8uxqPQTxEyWwDKw,45062
|
|
24
|
+
opentau/datasets/video_utils.py,sha256=NY20Et6SKWLdG4EjTNdXhpPqWEFON-UccIn_P2YukSQ,21810
|
|
25
|
+
opentau/datasets/grounding/__init__.py,sha256=ojvdscCIjp5EQxptFAjPgvjKGZa_Xk9LLZ2wNUebWFw,3139
|
|
26
|
+
opentau/datasets/grounding/base.py,sha256=FDAn2QPQHNBB7mzD7IQ2Bz882Dt1RPasBTgskXqKbP4,5773
|
|
27
|
+
opentau/datasets/grounding/clevr.py,sha256=lNZ0hr5EhQKTh-eg35zujybcAo5p8JQEn9dW8DJhOjI,3983
|
|
28
|
+
opentau/datasets/grounding/cocoqa.py,sha256=JpqVlI704tKSF-t5p-7588-wXGEWoEKrVKBxBLFn7mc,4429
|
|
29
|
+
opentau/datasets/grounding/dummy.py,sha256=YcO3Z7VF2tkPWYSs4U0vxypyPqgbPcFCxwpANgivLFA,3470
|
|
30
|
+
opentau/datasets/grounding/pixmo.py,sha256=Tm3LE8bGq93aeXqoF9W-JwJXt45_8Y86NUd9u15UJ5w,6251
|
|
31
|
+
opentau/datasets/grounding/vsr.py,sha256=c-wPaPVuukM8HKjGc6tF56TjxKMyZogE62VeUj-tTrs,4985
|
|
32
|
+
opentau/datasets/push_dataset_to_hub/utils.py,sha256=UqWfJzpg2NvSt43xYfTFmJvoFuwPfaVfMn55NXNIz6w,5070
|
|
33
|
+
opentau/datasets/v2/batch_convert_dataset_v1_to_v2.py,sha256=qF6Nzb0kgO6m3Ha-JnKpGLh0KOHqMb8Sv4SNDiyERic,43111
|
|
34
|
+
opentau/datasets/v2/convert_dataset_v1_to_v2.py,sha256=4sJw18EVC3UQaIEBzY0STKkdKgBHlT4Gy31SXKeEXhI,31654
|
|
35
|
+
opentau/datasets/v21/_remove_language_instruction.py,sha256=_5BWWWTAksQ9NyT7092LhkgoHItaPwOTH4MCGhZLNOk,4051
|
|
36
|
+
opentau/datasets/v21/batch_convert_dataset_v20_to_v21.py,sha256=wi2M5pHGZigLlfgcDKoi0B0HY5p7ue5-R2Nw-ArOs5Q,2104
|
|
37
|
+
opentau/datasets/v21/convert_dataset_v20_to_v21.py,sha256=_N5Gof3tCKqHd8X9mGHKWwgMi8iUQOqBXw6ysBzFdRw,5899
|
|
38
|
+
opentau/datasets/v21/convert_stats.py,sha256=5AollivnhmNEIzQ6EJQqk6KA09pKS_1mCfSvoObBOQA,6210
|
|
39
|
+
opentau/envs/__init__.py,sha256=ty4vilywul2t4vWcIHdmqli6gu5PGprGlwV9zdMwLBE,827
|
|
40
|
+
opentau/envs/configs.py,sha256=eIOOz7hjGrQYM_MhUca5aTr6gPFlM_NFIMjNt1GdxPo,7390
|
|
41
|
+
opentau/envs/factory.py,sha256=R1bh5QttjLllTn9aVaeK7C4f6Wt7vRERniQU72L6hqs,3749
|
|
42
|
+
opentau/envs/libero.py,sha256=UcsEXI7zVvKoy0yN0H_58r0sscyD4e8GKZt4An59Z7M,17777
|
|
43
|
+
opentau/envs/utils.py,sha256=4Ety8w9luWKmzyrG5wFJHfEOyQkBUvgkFCgdIu8FaHM,8014
|
|
44
|
+
opentau/optim/__init__.py,sha256=lry1B13y6AlLpryHytuIHAmDs8gMgOdWYduZ3JJCMUw,726
|
|
45
|
+
opentau/optim/factory.py,sha256=JmvkpPbMeX_VtiENCRsmJe0ekMucuV42eYAT-mggin0,1930
|
|
46
|
+
opentau/optim/optimizers.py,sha256=1Egb1hzAK37Ldy8-YZ8wzIgSpobdjbfhv6_eOGrUAQM,3877
|
|
47
|
+
opentau/optim/schedulers.py,sha256=SUIphJKqNhsylI-7-q_16E6YgiITU5DLqPKiZRUlVyY,5049
|
|
48
|
+
opentau/planner/__init__.py,sha256=nFKr6MR3lqJyj5EA2n8dabd-N4JZZNcp5KXJ42Yv9ls,3694
|
|
49
|
+
opentau/planner/high_level_planner.py,sha256=8FCWHyzP1pXbp-Dvn_txz0lWE-IywHGLlG1WMGXq9G0,12814
|
|
50
|
+
opentau/planner/utils/memory.py,sha256=AbyL-yK2BAhe35jngDyl01u1EugSz3snUGf8INlFm14,2288
|
|
51
|
+
opentau/planner/utils/utils.py,sha256=K9DHF9jPZAYMtSqltedMQrcQa8o2Zbq8tAwJG7QvzFk,2047
|
|
52
|
+
opentau/policies/__init__.py,sha256=KcyqKB-fbSG06g_pVjqAmymyRwgxfMJTPujGKhUY4Uw,998
|
|
53
|
+
opentau/policies/factory.py,sha256=fXAw6_Aij-mcdwU5SvN32MPl7fb6Z4ywSOASCDSntZw,6632
|
|
54
|
+
opentau/policies/normalize.py,sha256=gHOyf0RKGAOd-SbvdbHw2kGH498q5SWzDAelY0nKxJg,13846
|
|
55
|
+
opentau/policies/pretrained.py,sha256=JouuyJUdS_sh2inUKgcN7feUmszbXiN7CuLm2cBSluw,12836
|
|
56
|
+
opentau/policies/utils.py,sha256=AHvI2eRAJAEQvQUCIBgPIxj22DVV4Vx5spooqxImYDk,4367
|
|
57
|
+
opentau/policies/pi0/__init__.py,sha256=1JGQnTmfJLcog8ejL9JPp55JoY7FPx4dg57RWBIzqCA,817
|
|
58
|
+
opentau/policies/pi0/configuration_pi0.py,sha256=94EG2QlraDsPjD0zyuGwKPqToqV_ayPcUnx-JI-Bi7I,10329
|
|
59
|
+
opentau/policies/pi0/modeling_pi0.py,sha256=rz1S7hDOVEv12sN0ECGupddKwQVXMqdvm7G_OooWZLA,37442
|
|
60
|
+
opentau/policies/pi0/paligemma_with_expert.py,sha256=j9P6SL7MVP11MgyJqNnsrZAlOctWmqwDaJJAx4z9F84,20724
|
|
61
|
+
opentau/policies/pi05/__init__.py,sha256=VcIjZwlRW1JChRHqAK9Vz4JAIEP40RrP-W-UdyR6xk4,821
|
|
62
|
+
opentau/policies/pi05/configuration_pi05.py,sha256=ucgCC3BaIC6rcnotMYbElTq7ymPZh4xDGghxCseK33M,9307
|
|
63
|
+
opentau/policies/pi05/modeling_pi05.py,sha256=sF4OPGQWQ0eJevGDRZDPUItBkBC5Wp6WdpwsN4uqS14,50639
|
|
64
|
+
opentau/policies/pi05/paligemma_with_expert.py,sha256=nxBfUwBt6S4WwPDkn8LW3lHFcOib4CtoZzJEq4VKlck,23328
|
|
65
|
+
opentau/policies/value/__init__.py,sha256=wUP5vdpsvfnREqt4enfwaakzJ-ynX9sLYN14t2zEtpA,772
|
|
66
|
+
opentau/policies/value/configuration_value.py,sha256=ApjrNKHxvjNlSZ71-BPvanNAJh9GzAK9W0JCiz3mMHs,5951
|
|
67
|
+
opentau/policies/value/modeling_value.py,sha256=21x2EVGFlJbLasJDGTs3b5YMrrChrkuGxrYP7wLjkCY,18594
|
|
68
|
+
opentau/policies/value/reward.py,sha256=1ky4u5YCMTGneKs43550w8dZmBpKf4ZEMLt0cm8xpV0,3618
|
|
69
|
+
opentau/policies/value/siglip_gemma.py,sha256=9pyy4DvT_ewA0KnrQ9Wq_EPnK4fXPTlGp4_SyG1K9Lo,8147
|
|
70
|
+
opentau/scripts/actions_mse_loss.py,sha256=cc_VmN0atyOvkLlOsp4Atf9QzgCqG4_j9Z8z7yCjzSk,3216
|
|
71
|
+
opentau/scripts/bin_to_safetensors.py,sha256=yB-z27Rn3RVRlCWBY7-_yaDrVJ1qCgoasVvneapuui8,5288
|
|
72
|
+
opentau/scripts/compute_max_token_length.py,sha256=eZS6wAcTuPACk-G5B-PjGSaSJF9QIYj25GbzAf4AwNE,3976
|
|
73
|
+
opentau/scripts/display_sys_info.py,sha256=J174YJ3ccBknkZtjZrIS9p5pEAZ4qMbr5cEagmKTshM,2763
|
|
74
|
+
opentau/scripts/download_libero_benchmarks.py,sha256=c_ZggZjFzJCqvdUFliWxhmFB_3C5KR4s_SyLjEhc6Is,1848
|
|
75
|
+
opentau/scripts/eval.py,sha256=QpUdxLElwUZzDkZw3tJ-B2luxh2-P5UgRMNG7dlA3V0,34592
|
|
76
|
+
opentau/scripts/export_to_onnx.py,sha256=diyITonnGedv5Nm4eAz77BTEwoivOKYHvqYEG9Z5b08,6369
|
|
77
|
+
opentau/scripts/fake_tensor_training.py,sha256=y4F3CFs2jjpIJcT1wKvsrgFEebU9QFzbayhG2XT4UM0,3123
|
|
78
|
+
opentau/scripts/get_advantage_and_percentiles.py,sha256=JdjlADYzdS1Jc_19H6lLYMRnPlWxeckRSUQqwqb0rC4,8993
|
|
79
|
+
opentau/scripts/high_level_planner_inference.py,sha256=nbXr8Hp64YGeprMTpT8kvT_NgpBlI02CUlO6Mm2Js_E,3846
|
|
80
|
+
opentau/scripts/inference.py,sha256=_lp9YjPzarAnjiA8k2jBlIKZxza6PEHw--UyaqLPdNo,2110
|
|
81
|
+
opentau/scripts/launch_train.py,sha256=ThyZ0IqRfarvD3qEqa8sazSgYYL3Bh22zz6-z1JtZBs,2066
|
|
82
|
+
opentau/scripts/libero_simulation_parallel.py,sha256=qMee6T0EwMoAT1J2u8X4w8rsbOJYwyqD3LRAPe2Ta1g,13105
|
|
83
|
+
opentau/scripts/libero_simulation_sequential.py,sha256=xFSUQEuyai20QD-pYitp-UJPGE9zlaaIu4YSO0bhYKg,4775
|
|
84
|
+
opentau/scripts/nav_high_level_planner_inference.py,sha256=z2WHw68NWi-fJUd5TV4CrJHzxo-L7e2UliGjfOlqifM,1878
|
|
85
|
+
opentau/scripts/train.py,sha256=UftedMDTlTDLidV8fMPtiCu8xK9G16ivtIqaJp38su0,16866
|
|
86
|
+
opentau/scripts/visualize_dataset.py,sha256=_xGfAXQqhjGYMi__6L7qRH2xS5XQ2-GQRXjNw3KXMlY,10109
|
|
87
|
+
opentau/scripts/visualize_dataset_html.py,sha256=gEX-E5fFqBhINthf7xLMICHySvw9e3Kcf1HPRnJIyug,17979
|
|
88
|
+
opentau/scripts/zero_to_fp32.py,sha256=Rkl1ZczytKix9vGMg0EELzdJYFqUM1yB9p3xvSaK9k8,33272
|
|
89
|
+
opentau/utils/__init__.py,sha256=hIUeGPpZHf2AVf0-5C2p0BOcY0cFHCTT5yHn-SpEPwY,856
|
|
90
|
+
opentau/utils/accelerate_utils.py,sha256=vXnSGo1hXCUNof-oNKLMJ_SOMjpKhpZ1gx21ObSsopI,2630
|
|
91
|
+
opentau/utils/benchmark.py,sha256=jVli6gdBRMXAqNM3AIi43a0N_O1CLQMbKXsPK_e2y3s,3063
|
|
92
|
+
opentau/utils/fake_tensor.py,sha256=H8yIJJfO2vqrmWH9jXZkHqmndbd8WsO_2QTu0BZsAWY,2668
|
|
93
|
+
opentau/utils/hub.py,sha256=WGppA8j6P779u5eVWrkpmAfIJYoVzjLAKpeM8hQfm90,9319
|
|
94
|
+
opentau/utils/import_utils.py,sha256=bcfOUoW1wX6mAC-C5ESx0ABszDDIF_3PQp3uHe_gBWc,3429
|
|
95
|
+
opentau/utils/io_utils.py,sha256=z5S1MUF3PDOJrDmZz5FSs9FRxyiyjphLIvkUlkx4Jzs,5386
|
|
96
|
+
opentau/utils/libero.py,sha256=blV5ZMtxJ_ADUoLyeE_pIZCegzUcx_vOltzAWY1zXe8,7832
|
|
97
|
+
opentau/utils/libero_dataset_recorder.py,sha256=oafBuusXGqbMUqZrzoBrG5C441lyy8jBM3pKKwE_9rE,17024
|
|
98
|
+
opentau/utils/logging_utils.py,sha256=zd7ypmk7aqVposPhA7Kg-PYrstapY4MsuTklsTD4r4o,5989
|
|
99
|
+
opentau/utils/monkey_patch.py,sha256=cVgZ1N-NNVnlRKPA1dwO9FM4IbxR0V_Hbil6p-6knhA,9558
|
|
100
|
+
opentau/utils/random_utils.py,sha256=k3Ab3Y98LozGdsBzKoP8xSsFTcnaRqUzY34BsETCrrA,9102
|
|
101
|
+
opentau/utils/train_utils.py,sha256=0d7yvk8wlP-75pwB55gr095b_b1sWG5nlqdVxyH6_o0,6796
|
|
102
|
+
opentau/utils/utils.py,sha256=DrMStfjBEkw_8WVhYMnCQJNBxMeozIJ8LBSpOtMQhFM,15760
|
|
103
|
+
opentau-0.1.0.dist-info/licenses/LICENSE,sha256=tl3_NkxplsgU86xSvEWnDlE1UR_JsIvGo7t4hPtsIbE,27680
|
|
104
|
+
opentau-0.1.0.dist-info/METADATA,sha256=D89t5Nd6jD5bsHXc532C0yypVj2fOWYeThanhqj8Q_8,10456
|
|
105
|
+
opentau-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
106
|
+
opentau-0.1.0.dist-info/entry_points.txt,sha256=Q2Jf-g98RrhE7trmkvW3KrOgSBQJfdRRgUxaraF-A3c,68
|
|
107
|
+
opentau-0.1.0.dist-info/top_level.txt,sha256=7_yrS4x5KSeTRr2LICTCNOZmF-1_kSOFPKHvtJPL1Dw,8
|
|
108
|
+
opentau-0.1.0.dist-info/RECORD,,
|