.pth .tar.gz data/ MNIST/ cifar-10-batches-py/ experiment_data/