Avoid h being overwritten by residual p (#1)

vijaydwivedi75 · Jan 18, 2022 · d7f1f14 · d7f1f14
1 parent 21d2a65
commit d7f1f14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/layers/pna_lspe_layer.py b/layers/pna_lspe_layer.py
@@ -342,9 +342,9 @@ def forward(self, g, h, p, e, snorm_n):
 
         if self.residual:
             h = h_in + h  # residual connection
-            h = p_in + p  # residual connection
+            p = p_in + p  # residual connection
 
         return h, p
 
     def __repr__(self):
-        return '{}(in_channels={}, out_channels={})'.format(self.__class__.__name__, self.in_dim, self.out_dim)
+        return '{}(in_channels={}, out_channels={})'.format(self.__class__.__name__, self.in_dim, self.out_dim)