dgrachev commited on
Commit
0055e9a
·
verified ·
1 Parent(s): e9ecd43

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # A2 Pretrained Model
2
+
3
+ Pretrained ViLGP3D model for 6-DOF grasp pose selection in tabletop manipulation.
4
+
5
+ ## Model Architecture
6
+
7
+ - **Network**: CLIPAction (CLIP-based action selection with cross-attention)
8
+ - **Width**: 768
9
+ - **Layers**: 1
10
+ - **Heads**: 8
11
+ - **Action Dim**: 7 (xyz + quaternion)
12
+ - **Features**: RoPE (Rotary Position Encoding)
13
+
14
+ ## Usage
15
+
16
+ ```python
17
+ from lerobot_policy_a2 import A2Policy, A2Config
18
+
19
+ # Load pretrained model
20
+ policy = A2Policy.from_pretrained("dgrachev/a2_pretrained")
21
+ ```
22
+
23
+ ## Training Data
24
+
25
+ Trained on simulated tabletop grasping with UR5e robot and Robotiq gripper.
26
+
27
+ ## Related
28
+
29
+ - Environment: Install with `pip install lerobot[a2]`
30
+ - Assets: [dgrachev/a2_assets](https://huggingface.co/datasets/dgrachev/a2_assets)