Untitled
unknown
plain_text
3 years ago
742 B
13
Indexable
self.convs = torch.nn.ModuleList()
self.convs.append(
SAGEConv(
args.in_channels,
args.hidden_channels,
aggr=aggr,
aggr_kwargs=aggr_kwargs))
self.convs.append(
ClusterGCNConv(
args.hidden_channels,
args.hidden_channels
aggr_kwargs=aggr_kwargs))
self.convs.append(
ClusterGCNConv(
args.hidden_channels,
args.hidden_channels
aggr_kwargs=aggr_kwargs))
self.convs.append(
ClusterGCNConv(
args.hidden_channels,
args.hidden_channels
aggr_kwargs=aggr_kwargs,
))Editor is loading...