Skip to content

Commit e92ed01

Browse files
committed
fix gat and unigat bugs
1 parent 55df18c commit e92ed01

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

dhg/nn/convs/graphs/gat_conv.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,5 +59,4 @@ def forward(self, X: torch.Tensor, g: Graph) -> torch.Tensor:
5959
X = self.act(X)
6060
if self.bn is not None:
6161
X = self.bn(X)
62-
X = self.drop(X)
6362
return X

dhg/nn/convs/hypergraphs/unignn_conv.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,6 @@ def forward(self, X: torch.Tensor, hg: Hypergraph) -> torch.Tensor:
145145
X = self.act(X)
146146
if self.bn is not None:
147147
X = self.bn(X)
148-
X = self.drop(X)
149148
return X
150149

151150

0 commit comments

Comments
 (0)