IliaLarchenko/dot_pusht_images

Model Card for "Decoder Only Transformer (DOT) Policy" for PushT images dataset

Read more about the model and implementation details in the DOT Policy repository.

This model is trained using the LeRobot library and achieves state-of-the-art results on behavior cloning on the PushT images dataset. It achieves a 74.2% success rate (and 0.936 average max reward) vs. ~69% for the previous state-of-the-art model (Diffusion and VQ-BET perform the same).

This result is achieved without the checkpoint selection and is easy to reproduce.

You can use this model by installing LeRobot from this branch

To train the model:

python lerobot/scripts/train.py policy=dot_pusht_image env=pusht

To evaluate the model:

python lerobot/scripts/eval.py -p IliaLarchenko/dot_pusht_images eval.n_episodes=1000 eval.batch_size=100 seed=1000000

Model size:

• Total parameters: 14.1m
• Trainable parameters: 2.9m