Hugging Face's logo

SinjaeKang
/
wp_robomimic

Robotics
robomimic
imitation-learning
flow-matching
Model card Files
xet
Community
wp_robomimic
File size: 2,127 Bytes
beb2bc2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9070024
 
 
 
 
 
 
 
beb2bc2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
 
---
tags:
- robomimic
- robotics
- imitation-learning
- flow-matching
---

# WarmPrior Robomimic Checkpoints

This repository contains model checkpoints from the WarmPrior_Robomimic experiments.

## Repository Structure

Each experiment run is stored in its own folder with the following structure:

```
run_name/
├── model_best.pt   # Best performing checkpoint
└── model_latest.pt # Latest checkpoint
```

## Available Runs

| Run Name | Method | Best | Latest |
|----------|--------|------|--------|
| `PA_0116_010604_tool_hang_ph_image_flow_chitransformer_384_noise0.5_seed2` | PA | ✅ | ✅ |
| `PA_0116_010619_tool_hang_ph_image_flow_chiunet_256_noise0.5_seed2` | PA | ✅ | ✅ |
| `PA_0116_010624_square_mh_image_flow_chitransformer_384_noise0.5_seed2` | PA | ✅ | ✅ |
| `PA_0116_010629_square_mh_image_flow_chiunet_256_noise0.5_seed2` | PA | ✅ | ✅ |
| `PA_0117_141516_square_mh_image_flow_chiunet_256_noise0.3_seed2` | PA | ✅ | ✅ |
| `PA_0117_165708_square_mh_image_flow_chitransformer_384_noise0.3_seed2` | PA | ✅ | ✅ |
| `PA_0117_211126_tool_hang_ph_image_flow_chiunet_256_noise0.3_seed2` | PA | ✅ | ✅ |
| `PA_0117_231750_tool_hang_ph_image_flow_chitransformer_384_noise0.3_seed2` | PA | ✅ | ✅ |

## Usage

To download a specific checkpoint:

```python
from huggingface_hub import hf_hub_download
import torch

# Download a specific checkpoint
model_path = hf_hub_download(
    repo_id="SinjaeKang/wp_robomimic",
    filename="RUN_NAME/model_best.pt"
)

# Load the model
checkpoint = torch.load(model_path, map_location='cpu')
```

Or download all checkpoints:

```python
from huggingface_hub import snapshot_download

# Download entire repository
local_dir = snapshot_download(
    repo_id="SinjaeKang/wp_robomimic",
    local_dir="./checkpoints"
)
```

## Methods

- **BASE**: Baseline model without warm prior
- **PA**: Previous Action warm prior
- **AC**: Action Chunk warm prior

## Training Details

For more information about the training procedure and methodology, please refer to the [WarmPrior_Robomimic repository](https://github.com/YourUsername/WarmPrior_Robomimic).