Use neighbor sampling for batch training of large graphs from `OGB`

Update GNN trainer to use [torch_geometric.loader.NeighborLoader](https://pytorch-geometric.readthedocs.io/en/latest/modules/loader.html#torch_geometric.loader.NeighborLoader).