diff options
Diffstat (limited to 'models')
| -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), ) |