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