Changed to leaky relu

WouterBesse · May 16, 2023 · a0b2146 · a0b2146
1 parent 6306f1b
commit a0b2146
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/Tybalt/TybaltVAE.py b/models/Tybalt/TybaltVAE.py
@@ -5,7 +5,7 @@ def linear_block(input_size, output_size):
     block = nn.Sequential(
         nn.Linear(input_size, output_size),
         nn.BatchNorm1d(output_size),
-        nn.ReLU())
+        nn.LeakyReLU(0.1))
     return block
 
 class Decoder(nn.Module):