remove an activation function

lucidrains · Sep 4, 2023 · 9a364dd · 9a364dd
1 parent 333d731
commit 9a364dd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
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'