2214cce644
|
||
|---|---|---|
| configs | ||
| film_efficientnet | ||
| tokenizers | ||
| trained_checkpoints | ||
| .gitattributes | ||
| .gitignore | ||
| .pylintrc | ||
| CONTRIBUTING.md | ||
| LICENSE | ||
| README.md | ||
| __init__.py | ||
| requirements.txt | ||
| robotics_transformer.blueprint | ||
| sequence_agent.py | ||
| sequence_agent_test.py | ||
| sequence_agent_test_set_up.py | ||
| transformer.py | ||
| transformer_network.py | ||
| transformer_network_test.py | ||
| transformer_network_test_set_up.py | ||
| transformer_test.py | ||
README.md
Robotics Transformer
This is not an officially supported Google product.
This repository is a collection code files and artifacts for running Robotics Transformer or RT-1.
Features
- Film efficient net based image tokenizer backbone
- Token learner based compression of input tokens
- Transformer for end to end robotic control
- Testing utilities
Getting Started
Installation
Clone the repo
git clone https://github.com/google-research/robotics_transformer.git
pip install -r robotics_transformer/requirements.txt
python -m robotics_transformer.tokenizers.action_tokenizer.test
Running Tests
To run RT-1 tests, you can clone the git repo and run bazel:
git clone https://github.com/google_research/robotics_transformer.git
cd robotics_transformer
bazel test ...
Using trained checkpoints
Checkpoints are included in trained_checkpoints/ folder for three models:
- RT-1 trained on 700 tasks
- RT-1 jointly trained on EDR and Kuka data
- RT-1 jointly trained on sim and real data
They are tensorflow SavedModel files. Instructions on usage can be found here
Future Releases
The current repository includes an initial set of libraries for early adoption. More components may come in future releases.
License
The Robotics Transformer library is licensed under the terms of the Apache license.