add README
Browse files
README.md
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags: [jax, flax, cifar10, mnist, slice-sampling]
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
+
# slice-sampling checkpoints
|
| 7 |
+
|
| 8 |
+
Auxiliary checkpoints for the [slice_sampling](https://github.com/) research
|
| 9 |
+
codebase (`outsourced/` targets that build MCMC benchmarks over pretrained
|
| 10 |
+
generator + classifier stacks). Not general-purpose models.
|
| 11 |
+
|
| 12 |
+
| file | arch | dataset |
|
| 13 |
+
|------|------|---------|
|
| 14 |
+
| `cifar10_classifier.pkl` | `Cifar10CNN` (~1.15M params, GroupNorm, 32x32 native) | CIFAR10 |
|
| 15 |
+
| `mnist_classifier.pkl` | `Classifier` (2 conv stages + FC, 32x32) | MNIST |
|
| 16 |
+
| `mnist_vae.pkl` | ResNet-style VAE, `latent_dim=16` | MNIST |
|
| 17 |
+
|
| 18 |
+
Fetched automatically by `outsourced.common.load_checkpoint` when the local
|
| 19 |
+
file is missing.
|