NoahWeiss commited on
Commit
266bf23
·
verified ·
1 Parent(s): 37629a3

model card: FFT recipe + held-out split

Browse files
Files changed (1) hide show
  1. README.md +40 -0
README.md ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: physical-intelligence/pi05_base
4
+ datasets:
5
+ - NoahWeiss/farm_uf850_home_full
6
+ tags:
7
+ - robotics
8
+ - pi0.5
9
+ - openpi
10
+ - ufactory-uf850
11
+ ---
12
+
13
+ # farm_uf850_home_fft
14
+
15
+ Full fine-tune (every parameter trainable, no LoRA/GSE) of pi0.5 on
16
+ [NoahWeiss/farm_uf850_home_full](https://huggingface.co/datasets/NoahWeiss/farm_uf850_home_full)
17
+ (revision `v2.0`): 1153 teleoperated UFactory UF850 episodes across 20 home
18
+ manipulation tasks, recorded with a Quest 3 rig.
19
+
20
+ ## Held-out split
21
+
22
+ 63 episodes (2-5 per task, stratified, seed 153) were excluded from training
23
+ and reserved for evaluation. The full labeled list is in
24
+ [`heldout_home_full.json`](./heldout_home_full.json) in this repo. Training
25
+ used the remaining 1090 episodes (~377k frames). Norm stats were computed on
26
+ the train split only.
27
+
28
+ ## Recipe
29
+
30
+ - openpi `pi05` config `pi05_farm_home_fft`: absolute joint actions
31
+ (`use_delta_joint_actions=False`), `action_horizon=10`, EMA 0.999
32
+ (saved params are the EMA copy)
33
+ - 8x NVIDIA H200, `fsdp_devices=2` (4 data-parallel replicas), batch 64
34
+ - AdamW, cosine schedule: 2k warmup, peak LR 3.5e-5, decay to 3.5e-6
35
+ - heavy image domain randomization (brightness/contrast/saturation/hue,
36
+ per-channel gamma, grayscale, blur, crop/rotate)
37
+ - ~12 h single-node run; checkpoints streamed here every `save_interval`
38
+ steps as `step-<N>` tags (params + assets, inference-ready)
39
+
40
+ Training/eval code: the FARM repo (`model/amp_fft/`).