Use custom dataloader for training

[h3jia]

Hello, is there a straightforward way to use custom pytorch DataLoader for training, instead of the append_simulations method? I didn't find this information in the documentation. This will allow us to add some additional data augmentation when loading the data. Thanks.