harrywang01 commited on
Commit
009cf34
·
verified ·
1 Parent(s): 9443932

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +31 -0
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags: [diffusion-policy, robotics, franka]
4
+ ---
5
+ # oopsie-debug
6
+
7
+ Diffusion Policy (image, UNet) checkpoints trained on the Franka teleop block dataset
8
+ `teleop_dataset_20260920_205906` (20 episodes, 1206 steps @ 10 Hz, all demos used for training,
9
+ no held-out validation). Trained 2001 epochs (~34k steps, batch 64); loss values in the table are
10
+ in-sample train loss.
11
+
12
+ | file | epoch | train loss |
13
+ |---|---|---|
14
+ | epoch_0500.ckpt | 500 | 0.0072 |
15
+ | epoch_1000.ckpt | 1000 | 0.0031 |
16
+ | epoch_1500.ckpt | 1500 | 0.0011 |
17
+ | epoch_2000.ckpt | 2000 | 0.0004 |
18
+
19
+ Each `.ckpt` is a full training checkpoint (`state_dicts`: model, ema_model, optimizer;
20
+ `pickles`: epoch, global_step, rng_state; `cfg`), loadable with
21
+ `torch.load(path, pickle_module=dill, map_location='cpu', weights_only=False)`.
22
+ Use `state_dicts['ema_model']` for inference.
23
+
24
+ ## Observation / action contract
25
+ - obs: `img_third`, `img_wrist` (3x120x160 RGB, resized from 480x640 BGR source), `robot_eef_pose`
26
+ (9-D: xyz + rotation_6d), `gripper_state` (1-D width in m). n_obs_steps = 2.
27
+ - action (10-D): `[x, y, z, rot6d(6), gripper]` = pose/gripper of frame t+1. horizon 16, 8 executed.
28
+ - Normalization: position min/max -> [-1,1]; rotation_6d identity (no min/max);
29
+ gripper fixed [0, 0.08] m -> [-1,1]. The source quaternion is scalar-last (xyzw).
30
+ - Code: `memory_diffusion_policy/dataset/teleop_block_image_dataset.py`
31
+ (`action10_to_pose7_gripper` is the inverse for rollout). Full config in `config.yaml`.