To train a model for CIFAR-10 or other datasets, run:
python main.py --dataset=[cifar10] --name=resnet_[method] --epochs=400 --is_train=True
cifar10 should be one of [cifar10, svhn, fashion-mnist, AffNIST, CIFAR-100, MNIST]. method should be one of [global_routing, em_routing].
To test a model, simply run:
python main.py --dataset=[cifar10] --name=resnet_[method] --epochs=400 --is_train=False
cifar10 should be one of [cifar10, svhn, fashion-mnist, AffNIST, CIFAR-100, MNIST]. method should be one of [global_routing, em_routing].