This is the implementation of the 2-GNN for kg.

# Reproduce 2-WL local 
```
python train.py --dataset GraIL-BM_fb237_v1 --test 
```

# Reproduce 2-FWL local
```
python train_fwl.py --dataset GraIL-BM_fb237_v1 --test 
```