commabody-gpt2 / README.md
armandpl's picture
Create README.md
fe22f1a verified
|
raw
history blame
416 Bytes
---
datasets:
- commaai/commabody
pipeline_tag: robotics
---
This model has been trained on a larger version of the commabody dataset. It includes a vqgan gan encoder/decoder fine tuned from imagenet, as well as a GPT2 trained to predict the next frame, wheel speeds and actions.
It can be used either as a simulator or as a policy.
todo: link to blog post, link to body-jim example script, include example video