commabody-gpt2 / README.md
armandpl's picture
Create README.md
fe22f1a verified
|
raw
history blame
416 Bytes
metadata
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