mmacosha commited on
Commit
70ff252
·
verified ·
1 Parent(s): 29502a9

update README for per-model layout

Browse files
Files changed (1) hide show
  1. README.md +9 -6
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 'outsourced' experiments of Generalised Latent Slice Sampling progect.
 
 
 
9
 
10
- | file | arch | dataset |
11
- |------|------|---------|
12
- | `cifar10_classifier.pkl` | `Cifar10CNN` (~1.15M params, GroupNorm, 32x32 native) | CIFAR10 |
13
- | `mnist_classifier.pkl` | `Classifier` (2 conv stages + FC, 32x32) | MNIST |
14
- | `mnist_vae.pkl` | ResNet-style VAE, `latent_dim=16` | MNIST |
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.