def METHOD_NAME(self, x, adj_t, batch):
    x = self.atom_encoder(x)
    edge_attr = adj_t.coo()[2]
    adj_t = adj_t.set_value(self.bond_encoder(edge_attr), layout='coo')
    for conv in self.convs:
        x = conv(x, adj_t)
        x = F.dropout(x, p=self.dropout, training=self.training)
    x = global_mean_pool(x, batch)
    x = self.lin(x)
    return x