Skip to content

Commit

Permalink
fix v2 param description
Browse files Browse the repository at this point in the history
  • Loading branch information
mattjhayes3 committed Dec 9, 2024
1 parent 95b1e57 commit a863103
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch_geometric/nn/conv/gatv2_conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,8 +114,7 @@ class GATv2Conv(MessagePassing):
residual (bool, optional): If set to :obj:`True`, the layer will add
a learnable skip-connection. (default: :obj:`False`)
interactive_attn (bool, optional): If set to :obj:`False`, fixes
:math:`\mathbf{a}^{\top}_{t} \mathbf{\Theta}_{t}\mathbf{x}_i = 0`.
(default :obj:`True`)
:math:`\mathbf{\Theta}_{t}\mathbf{x}_i = 0`. (default :obj:`True`)
**kwargs (optional): Additional arguments of
:class:`torch_geometric.nn.conv.MessagePassing`.
Expand Down

0 comments on commit a863103

Please sign in to comment.