diff options
| author | gdamms <damguillotin@gmail.com> | 2026-02-05 19:55:58 +0100 |
|---|---|---|
| committer | gdamms <damguillotin@gmail.com> | 2026-02-05 19:55:58 +0100 |
| commit | f1be02af8c33d4136ad7c90dd43158a8ff6e5af1 (patch) | |
| tree | 3d7facac2732046ce1447156afe7dacc3b64d585 /models/autoencoder.py | |
| parent | a5d5f30fbd9c6c7c78834072401932c84bddaf14 (diff) | |
| download | diffusion-mnist-f1be02af8c33d4136ad7c90dd43158a8ff6e5af1.tar.gz diffusion-mnist-f1be02af8c33d4136ad7c90dd43158a8ff6e5af1.zip | |
better plots, semi working autoencoder
Diffstat (limited to 'models/autoencoder.py')
| -rw-r--r-- | models/autoencoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/autoencoder.py b/models/autoencoder.py index 353ec89..abbc3be 100644 --- a/models/autoencoder.py +++ b/models/autoencoder.py @@ -60,7 +60,7 @@ class Autoencoder(AEModule): nn.Conv2d(64, latent_channels, kernel_size=3, padding=1), nn.ReLU(), - # 7x7 -> 8x8 (no activation - latent space should be unconstrained) + # 7x7 -> 8x8 nn.Conv2d(latent_channels, latent_channels, kernel_size=2, padding=1), ) |