"tests/vscode:/vscode.git/clone" did not exist on "ebf505ee926f635480e64041bded9d161a5ef9ab"
Unverified Commit 76ce14b7 authored by xiangyuzhi's avatar xiangyuzhi Committed by GitHub
Browse files

[Sparse] Fix example variable name (#6581)

parent 46d7b1d9
...@@ -42,15 +42,15 @@ class SAGEConv(nn.Module): ...@@ -42,15 +42,15 @@ class SAGEConv(nn.Module):
def __init__( def __init__(
self, self,
in_feats, in_size,
out_feats, out_size,
): ):
super(SAGEConv, self).__init__() super(SAGEConv, self).__init__()
self._in_src_feats, self._in_dst_feats = in_feats, in_feats self._in_src_feats, self._in_dst_feats = in_size, in_size
self._out_feats = out_feats self._out_size = out_size
self.fc_neigh = nn.Linear(self._in_src_feats, out_feats, bias=False) self.fc_neigh = nn.Linear(self._in_src_feats, out_size, bias=False)
self.fc_self = nn.Linear(self._in_dst_feats, out_feats, bias=True) self.fc_self = nn.Linear(self._in_dst_feats, out_size, bias=True)
self.reset_parameters() self.reset_parameters()
def reset_parameters(self): def reset_parameters(self):
......
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