Efficient GNN Training

Improve efficiency of GNN Training