Update the example scripts to use the new `GraphNeTDataModule`

Created by: RasmusOrsoe

Since #642 we've gotten a much more elegant way of producing dataloaders for training and inference. We should update our training scripts to use this new method instead, and strongly consider to remove the old method, as discussed in #467.