robotics-diffusion-transformer
/

rdt-170m

Inference Endpoints

Model card Files Files and versions Community

robotics-diffusion-transformer commited on Oct 24

Commit

8aa386c

•

1 Parent(s): 7fd572e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ tags:
 ---
 # RDT-170M
- RDT-170M is a 170M-parameter imitation learning Diffusion Transformer ***(RDT(small) in ablation)***. Given language instruction and RGB images of up to three views, RDT can predict the next
 64 robot actions. RDT is compatible with almost all modern mobile manipulators, from single-arm to dual-arm, joint to EEF, position to velocity, and even wheeled locomotion.
  All the [code](https://github.com/thu-ml/RoboticsDiffusionTransformer/tree/main?tab=readme-ov-file), pre-trained model weights, and [data](https://huggingface.co/datasets/robotics-diffusion-transformer/rdt-ft-data) are licensed under the MIT license.

 ---
 # RDT-170M
+ RDT-170M is a 170M-parameter imitation learning Diffusion Transformer **(RDT(small) in ablation)**. It has a hidden size of `1024` and a depth of `14`, which are half of that in RDT-1B. Given language instruction and RGB images of up to three views, RDT can predict the next
 64 robot actions. RDT is compatible with almost all modern mobile manipulators, from single-arm to dual-arm, joint to EEF, position to velocity, and even wheeled locomotion.
  All the [code](https://github.com/thu-ml/RoboticsDiffusionTransformer/tree/main?tab=readme-ov-file), pre-trained model weights, and [data](https://huggingface.co/datasets/robotics-diffusion-transformer/rdt-ft-data) are licensed under the MIT license.