pip install -r requirements.txt
MNIST: 1x28x28, automatically downloaded from torchvision
CIFAR10: 3x32x32, resized as 3x28x28, automatically downloaded from torchvision
EMNIST: 1x28x28, automatically downloaded from torchvision
Sent140: manually download from https://www.kaggle.com/kazanova/sentiment140/download
-
install the requirements
-
change the parameters in
config.yml
-
python3 main.py