Update README.md
Browse files
README.md
CHANGED
|
@@ -11,8 +11,8 @@ tags:
|
|
| 11 |
# STIP tutorial checkpoints
|
| 12 |
|
| 13 |
Small checkpoints used by the [`stip`](https://github.com/instadeepai/stip) tutorial
|
| 14 |
-
notebooks
|
| 15 |
-
|
| 16 |
trained for 1500 steps on a 4-component 2D Gaussian mixture) and have no
|
| 17 |
use outside the notebooks.
|
| 18 |
|
|
@@ -54,13 +54,3 @@ gen_model = nnx.merge(graphdef, checkpointer.restore_ema(params))
|
|
| 54 |
|
| 55 |
`restore_ema` reads only `ema_params` and `extra`, and applies the same bias
|
| 56 |
correction the training loop uses for evaluation.
|
| 57 |
-
|
| 58 |
-
## Reproducing
|
| 59 |
-
|
| 60 |
-
```bash
|
| 61 |
-
uv run python tutorials/scripts/train_conditioning_checkpoints.py
|
| 62 |
-
```
|
| 63 |
-
|
| 64 |
-
The script mirrors the notebook's model definitions and PRNG chain, so it
|
| 65 |
-
reproduces these exact weights. A checkpoint pins the parameter structure: if a
|
| 66 |
-
notebook's network changes, re-run the script and re-upload.
|
|
|
|
| 11 |
# STIP tutorial checkpoints
|
| 12 |
|
| 13 |
Small checkpoints used by the [`stip`](https://github.com/instadeepai/stip) tutorial
|
| 14 |
+
notebooks to demonstrate sampling methods and capapbilities of the repository.
|
| 15 |
+
They are toy models (a two-layer MLP, ~50k parameters,
|
| 16 |
trained for 1500 steps on a 4-component 2D Gaussian mixture) and have no
|
| 17 |
use outside the notebooks.
|
| 18 |
|
|
|
|
| 54 |
|
| 55 |
`restore_ema` reads only `ema_params` and `extra`, and applies the same bias
|
| 56 |
correction the training loop uses for evaluation.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|