Downloads · 30 days
0
Himanshu77275/dit-diffusion-xarm-wire-rope-v0
dit-diffusion-xarm-wire-rope-v0 is a robotics model from Himanshu77275. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for lerobot.
This repository contains a Multi-Task DiT diffusion policy fine-tuned with LeRobot on the JayCao99/xarm-wire-rope-v0 dataset.
Downloads · 30 days
0
Access
Public
Updated Aug 4, 2026
Repo size
1.3 GB
Likes
1
Public
Click a slice to open those files.
.safetensors1.3 GB · 100%
From the Hugging Face model README
This repository contains a Multi-Task DiT diffusion policy fine-tuned with LeRobot on the JayCao99/xarm-wire-rope-v0 dataset.
| Field | Value |
|---|---|
| Policy | Multi-Task DiT |
| Objective | Diffusion |
| Dataset | JayCao99/xarm-wire-rope-v0 |
| Robot | xArm7 |
| Episodes | 210 |
| Frames | 290,153 |
| Cameras | ego, wrist |
| State dim | 15 |
| Action dim | 8 |
| Training steps | 100,000 |
| Batch size | 32 |
| Save frequency | 5,000 steps |
| GPU | 1x A100 80GB, Unity superpod-a100 |
| Runtime | approximately 8h 23m |
| Peak VRAM from log | approximately 21.72 GB |
| WandB | disabled |
| Checkpoint | Notes |
|---|---|
checkpoint-100000/ | Final policy checkpoint after 100k finetuning steps |
from huggingface_hub import snapshot_download
ckpt_dir = snapshot_download(
repo_id="Himanshu77275/dit-diffusion-xarm-wire-rope-v0",
allow_patterns="checkpoint-100000/*",
)
The checkpoint directory contains the LeRobot policy config, model weights, and dataset normalization/preprocessing files needed for policy loading.