increase size of fc-encoder

LarsKue · Jan 6, 2024 · a46bb02 · a46bb02
1 parent b6b10ce
commit a46bb02
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/models/model.py b/src/models/model.py
@@ -101,12 +101,9 @@ def __init__(self, hparams, *datasets):
             pools.Mean(dim=1),
             nn.Flatten(),
 
-            ResidualBlock(1024, 8),
-
-            nn.Sequential(nn.Linear(1024, 512), nn.SELU()),
-            ResidualBlock(512, 4),
+            ResidualBlock(1024, 12),
 
-            nn.Linear(512, self.hparams.embeddings),
+            nn.Linear(1024, self.hparams.embeddings),
         )
 
         self.flow = nn.Sequential(