-
Notifications
You must be signed in to change notification settings - Fork 27
Open
Description
for the code:
for _ in range(num_layers - 2):
self.convs.append(GATConv(hidden_channels*layer_heads[i-1], hidden_channels, heads=layer_heads[i], concat=True))
if self.batchnorm:
self.bns.append(torch.nn.BatchNorm1d(hidden_channels*layer_heads[i-1]))
the _ should be i
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels