fix gat and unigat bugs

iMoonLab · Sep 16, 2023 · e92ed01 · e92ed01
1 parent 55df18c
commit e92ed01
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/dhg/nn/convs/graphs/gat_conv.py b/dhg/nn/convs/graphs/gat_conv.py
@@ -59,5 +59,4 @@ def forward(self, X: torch.Tensor, g: Graph) -> torch.Tensor:
             X = self.act(X)
             if self.bn is not None:
                 X = self.bn(X)
-            X = self.drop(X)
         return X
diff --git a/dhg/nn/convs/hypergraphs/unignn_conv.py b/dhg/nn/convs/hypergraphs/unignn_conv.py
@@ -145,7 +145,6 @@ def forward(self, X: torch.Tensor, hg: Hypergraph) -> torch.Tensor:
             X = self.act(X)
             if self.bn is not None:
                 X = self.bn(X)
-            X = self.drop(X)
         return X