remove an activation function

lucidrains · lucidrains · commit 9a364dd33cf0 · 2023-09-04T15:52:05.000-07:00
diff --git a/gigagan_pytorch/gigagan_pytorch.py b/gigagan_pytorch/gigagan_pytorch.py
@@ -1543,7 +1543,6 @@ def __init__(
 
         self.to_logits = nn.Sequential(
             conv2d_3x3(dim_last, dim_last),
-            leaky_relu(),
             Rearrange('b c h w -> b (c h w)'),
             nn.Linear(dim_last * (4 ** 2), 1),
             Rearrange('b 1 -> b')
diff --git a/gigagan_pytorch/version.py b/gigagan_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.2.15'
+__version__ = '0.2.16'
