90e643c7a3727afef2caf932281e6ae91d62aac5,examples/cluster_gcn_ppi.py,Net,forward,#Net#Any#Any#,36

Before Change

    def forward(self, x, edge_index):
        x = F.elu(self.conv1(x, edge_index) + self.lin1(x))
        x = F.elu(self.conv2(x, edge_index) + self.lin2(x))
        x = self.conv3(x, edge_index) + self.lin3(x)
        return x


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

After Change

            x = batch_norm(x)
            x = F.relu(x)
            x = F.dropout(x, p=0.2, training=self.training)
        return self.convs[-1](x, edge_index)


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

Instances