Untitled

unknown
plain_text
2 years ago
742 B
Indexable
Never
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,
        ))