Hi Shuai,
I find the code in gnn.py takes 1 record per batch, will it suffers the problem of overfitting or takes more time in training and inference? And what is your motivation for this design.
Anyway, I learned a lot from your implementation. Thanks!