Untitled

Anonymous
plain_text
03/24/2023 9:16 AM
992 B
27
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...