Errata in the readme?
In the MNIST example in the readme, the functions tng_dataloader, val_dataloader, and test_dataloader return the same object. I think that at least for test, the train flag used to create the MNIST object should be False. I am not aware of how to extract automatically a validation dataset from the MNIST dataloader in torch.