Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Sep 21, 2022
1 parent 2778efc commit 6801ffa
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torch_geometric/nn/conv/hgt_conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,6 @@ def forward(

out = self.a_lin[node_type](F.gelu(out))
if out.size(-1) == x_dict[node_type].size(-1):

alpha = self.skip[node_type].sigmoid()
out = alpha * out + (1 - alpha) * x_dict[node_type]
out_dict[node_type] = out
Expand Down

0 comments on commit 6801ffa

Please sign in to comment.