update README for per-model layout
Browse files
README.md
CHANGED
|
@@ -5,13 +5,16 @@ tags: [jax, flax, cifar10, mnist, slice-sampling]
|
|
| 5 |
|
| 6 |
# slice-sampling checkpoints
|
| 7 |
|
| 8 |
-
Auxiliary checkpoints for the
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
-
|
|
| 11 |
-
|------|------|---------|
|
| 12 |
-
| `cifar10_classifier
|
| 13 |
-
| `mnist_classifier
|
| 14 |
-
| `mnist_vae
|
| 15 |
|
| 16 |
Fetched automatically by `outsourced.common.load_checkpoint` when the local
|
| 17 |
file is missing.
|
|
|
|
| 5 |
|
| 6 |
# slice-sampling checkpoints
|
| 7 |
|
| 8 |
+
Auxiliary checkpoints for the `slice_sampling` research codebase (`outsourced/`
|
| 9 |
+
targets that build MCMC benchmarks over pretrained generator + classifier
|
| 10 |
+
stacks). Each subdirectory is self-contained: the checkpoint, the model source
|
| 11 |
+
file, and a small `load.py` that reconstructs the model from the pickled meta.
|
| 12 |
|
| 13 |
+
| directory | model | dataset | notes |
|
| 14 |
+
|-----------|-------|---------|-------|
|
| 15 |
+
| `cifar10_classifier/` | `Cifar10CNN` (~1.15M params, GroupNorm) | CIFAR10 | 32x32 native, no upsample |
|
| 16 |
+
| `mnist_classifier/` | `Classifier` (2 conv stages + FC) | MNIST | 32x32 (padded) |
|
| 17 |
+
| `mnist_vae/` | ResNet-style VAE, `latent_dim=16` | MNIST | encoder/decoder both included |
|
| 18 |
|
| 19 |
Fetched automatically by `outsourced.common.load_checkpoint` when the local
|
| 20 |
file is missing.
|