Skip to content

Commit

Permalink
Update conv.py (dmlc#818)
Browse files Browse the repository at this point in the history
YichengDWu authored and yzh119 committed Sep 1, 2019
1 parent a9a0592 commit c4989b4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/dgl/nn/pytorch/conv.py
Original file line number Diff line number Diff line change
@@ -184,7 +184,7 @@ class GATConv(nn.Module):
.. math::
\alpha_{ij}^{l} & = \mathrm{softmax_i} (e_{ij}^{l})
e_{ij}^{l} & = \mathrm{LeakyReLU}\left(\vec{a}^T [W h^{I} \| W h^{j}]\right)
e_{ij}^{l} & = \mathrm{LeakyReLU}\left(\vec{a}^T [W h_{i} \| W h_{j}]\right)
Parameters
----------
@@ -672,7 +672,7 @@ class SAGEConv(nn.Module):
bias : bool
If True, adds a learnable bias to the output. Default: ``True``.
norm : callable activation function/layer or None, optional
If not None, applies normalization oto the updated node features.
If not None, applies normalization to the updated node features.
activation : callable activation function/layer or None, optional
If not None, applies an activation function to the updated node features.
Default: ``None``.

0 comments on commit c4989b4

Please sign in to comment.