From f1be02af8c33d4136ad7c90dd43158a8ff6e5af1 Mon Sep 17 00:00:00 2001
From: gdamms <damguillotin@gmail.com>
Date: Thu, 5 Feb 2026 19:55:58 +0100
Subject: better plots, semi working autoencoder

---
 models/autoencoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/autoencoder.py')

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),
         )
 
-- 
cgit v1.3.1