Requirements
Hardware environment: Intel(R) Xeon(R) Gold 6230R CPU @ 2.10GHz, NVIDIA GeForce RTX 3090 with 24GB memory.
Software environment: Ubuntu 18.04.6, Python 3.9, PyTorch 1.11.0 and CUDA 11.8.
- Please refer to PyTorch and PyG to install the environments;
- Run 'pip install -r requirements.txt' to download required packages;
Training
To train the model(s) in the paper
-
Please unzip datasets to the current file directory location
-
Open main.py to train AdaFGL with our proposed decouple two step training framework.
Run this command:
python main.py