Clarify fine-tuning and download path
Browse files
README.md
CHANGED
|
@@ -32,19 +32,19 @@ checkpoints/unified_encoder.ckpt model weights (EMA, bf16, 55.6M parameters)
|
|
| 32 |
```
|
| 33 |
|
| 34 |
Keep `.hydra/` next to `checkpoints/`, because the code rebuilds the model from it. The checkpoint
|
| 35 |
-
holds weights only, without optimizer state, so it supports evaluation and fine-tuning
|
| 36 |
-
|
| 37 |
|
| 38 |
## Quick Usage
|
| 39 |
|
| 40 |
With the [code](https://github.com/unified-inertial-encoder/Unified-Inertial-Encoder) installed:
|
| 41 |
|
| 42 |
```bash
|
| 43 |
-
|
|
|
|
| 44 |
hf download Open-IO/humanoid-g1-odometry --repo-type dataset \
|
| 45 |
--local-dir "$FOUNDATIONIMU_CACHE_ROOT/humanoid_superodom_clips_v2"
|
| 46 |
|
| 47 |
-
RUN=ckpts/unified-inertial-encoder
|
| 48 |
python -m src.eval_foundation \
|
| 49 |
--ckpt "$RUN/checkpoints/unified_encoder.ckpt" \
|
| 50 |
--data_from "$RUN" \
|
|
|
|
| 32 |
```
|
| 33 |
|
| 34 |
Keep `.hydra/` next to `checkpoints/`, because the code rebuilds the model from it. The checkpoint
|
| 35 |
+
holds weights only, without optimizer state, so it supports evaluation and fine-tuning
|
| 36 |
+
(`'meta.resume_from=<CKPT>' meta.resume_full_state=false`), not exact resumption of the training run.
|
| 37 |
|
| 38 |
## Quick Usage
|
| 39 |
|
| 40 |
With the [code](https://github.com/unified-inertial-encoder/Unified-Inertial-Encoder) installed:
|
| 41 |
|
| 42 |
```bash
|
| 43 |
+
RUN="${FOUNDATIONIMU_EXP_ROOT:-experiments}/unified-inertial-encoder"
|
| 44 |
+
hf download Open-IO/Unified-Inertial-Encoder --local-dir "$RUN"
|
| 45 |
hf download Open-IO/humanoid-g1-odometry --repo-type dataset \
|
| 46 |
--local-dir "$FOUNDATIONIMU_CACHE_ROOT/humanoid_superodom_clips_v2"
|
| 47 |
|
|
|
|
| 48 |
python -m src.eval_foundation \
|
| 49 |
--ckpt "$RUN/checkpoints/unified_encoder.ckpt" \
|
| 50 |
--data_from "$RUN" \
|