maximeseince commited on
Commit
95c1dc8
·
verified ·
1 Parent(s): 6011839

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -12
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, so that a tutorial can demonstrate sampling without spending ten
15
- minutes training first. 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,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.