Official codebase for paper Learning a Mini-batch Graph Transformer via Two-stage Interaction Augmentation.

See requirment.txt file for more information about how to install the dependencies.

We provide scripts to replicate the results in the paper.

sh run.sh

Table R1: The running times for large-scale datasets were recorded. The reported times represent the model training time for a single epoch, measured in seconds.