pytorch-geometric model has no configuration that allows to train for any number of classes.
pytorch-geometric model has no configuration that allows to train for any number of classes.