Robotics
LeRobot
Safetensors
diffusion
Eshwar-2123 commited on
Commit
45d8c18
·
verified ·
1 Parent(s): b23aef4

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Diffusion Policy for Pick-and-Place (SO-100)
2
+
3
+ ## Overview
4
+ This model is a Diffusion Policy trained using Hugging Face LeRobot for a single-task pick-and-place manipulation problem.
5
+
6
+ ## Robot
7
+ - SO-100 follower arm
8
+ - Front camera
9
+ - Wrist camera
10
+
11
+ ## Task
12
+ Pick the blue cube from one of four predefined locations and place it into a fixed target box.
13
+
14
+ ## Dataset
15
+ - 40 teleoperated demonstrations
16
+ - Dual-camera observations
17
+ - Dataset:
18
+ Eshwar-2123/pick_place_blue_cube_box_clean_v1_20260706_125359
19
+
20
+ ## Training
21
+ - Policy: Diffusion Policy
22
+ - Steps: 30,000
23
+ - Batch size: 16
24
+ - GPU: NVIDIA RTX 5080
25
+
26
+ ## Results
27
+ - Lowest validation loss: 0.037 (10k steps)
28
+ - Final training loss: 0.0036
29
+ - Final validation loss: 0.1303
30
+
31
+ ## Usage
32
+
33
+ ```python
34
+ from lerobot.policies import DiffusionPolicy