Hugging Face's logo

zou-research-lab
/
wjepa

Model card Files
xet
 Community
wjepa
File size: 3,111 Bytes
57300b0
 
 
b5b2874
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
---
license: apache-2.0
---

# WJEPA Checkpoints

This repository contains WJEPA checkpoints from a single seed for six datasets. Each dataset has two checkpoint variants: `participant` and `random`.

These checkpoints accompany the [W-JEPA code repository](https://github.com/Yogesh914/wjepa), which contains the model implementation, training pipeline, and inference scripts. See its README for setup and usage instructions.

There are 12 checkpoint files in total, approximately 744 MB.

## Folder structure

```text
.
├── README.md
├── dsads/
│   ├── participant/latest_checkpoint.pth
│   └── random/latest_checkpoint.pth
├── mhealth/
│   ├── participant/latest_checkpoint.pth
│   └── random/latest_checkpoint.pth
├── pamap2/
│   ├── participant/latest_checkpoint.pth
│   └── random/latest_checkpoint.pth
├── realworld/
│   ├── participant/latest_checkpoint.pth
│   └── random/latest_checkpoint.pth
├── uci_har/
│   ├── participant/latest_checkpoint.pth
│   └── random/latest_checkpoint.pth
└── usc_had/
    ├── participant/latest_checkpoint.pth
    └── random/latest_checkpoint.pth
```

Checkpoint paths follow `<dataset>/<variant>/latest_checkpoint.pth`. Select the dataset folder and either the `participant` or `random` variant to download the corresponding weights.

## Download the weights

### Install and authenticate

Install the Hugging Face Hub package, which includes the `hf` command:

```bash
python -m pip install -U huggingface_hub
```

For private repository access, log in with a Hugging Face account that has access to this repository. If using an access token, it needs read permission for `zou-research-lab/wjepa`.

```bash
hf auth login
```

### Download all checkpoints

```bash
hf download zou-research-lab/wjepa \
  --include "*.pth" \
  --local-dir ./wjepa
```

This saves all 12 checkpoints under `./wjepa/`, preserving the dataset and variant folders.

### Download one dataset

For example, download both UCI HAR variants:

```bash
hf download zou-research-lab/wjepa \
  --include "uci_har/*/*.pth" \
  --local-dir ./wjepa
```

Replace `uci_har` with any dataset folder listed above.

### Download one checkpoint

For example, download the UCI HAR `participant` checkpoint:

```bash
hf download zou-research-lab/wjepa \
  uci_har/participant/latest_checkpoint.pth \
  --local-dir ./wjepa
```

The file will be saved at `./wjepa/uci_har/participant/latest_checkpoint.pth`.

### Download with Python

After authenticating as described above, download a checkpoint with `hf_hub_download`:

```python
from huggingface_hub import hf_hub_download

checkpoint_path = hf_hub_download(
    repo_id="zou-research-lab/wjepa",
    filename="uci_har/participant/latest_checkpoint.pth",
    local_dir="./wjepa",
)

print(checkpoint_path)
```

To download all checkpoints from Python:

```python
from huggingface_hub import snapshot_download

snapshot_download(
    repo_id="zou-research-lab/wjepa",
    allow_patterns="*.pth",
    local_dir="./wjepa",
)
```