Skip to content

GitLab

Unverified Commit 50492d56 authored by V.E's avatar V.E Committed by GitHub
Browse files

Fixing issue #2921 (#2922) 

* fixing #2921

* fixing #2921
parent c0184365
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/nn/pytorch/conv/gatconv.py
View file @ 50492d56
...@@ -168,7 +168,7 @@ class GATConv(nn.Module): ...@@ -168,7 +168,7 @@ class GATConv(nn.Module):
else: else:
self.register_buffer('bias', None) self.register_buffer('bias', None)
if residual: if residual:
if self._in_dst_feats != out_feats: if self._in_dst_feats != out_feats * num_heads:
self.res_fc = nn.Linear( self.res_fc = nn.Linear(
self._in_dst_feats, num_heads * out_feats, bias=False) self._in_dst_feats, num_heads * out_feats, bias=False)
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment