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):
def __init__(
self,
in_feats,
out_feats,
in_size,
out_size,
):
super(SAGEConv, self).__init__()
self._in_src_feats, self._in_dst_feats = in_feats, in_feats
self._out_feats = out_feats
self._in_src_feats, self._in_dst_feats = in_size, in_size
self._out_size = out_size
self.fc_neigh = nn.Linear(self._in_src_feats, out_feats, bias=False)
self.fc_self = nn.Linear(self._in_dst_feats, out_feats, bias=True)
self.fc_neigh = nn.Linear(self._in_src_feats, out_size, bias=False)
self.fc_self = nn.Linear(self._in_dst_feats, out_size, bias=True)
self.reset_parameters()
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