This repository is the demo of GAS.
To install the required packages:
pip install -r requirements.txt
To train and evaluate the model(s) in the paper, run this command:
python GAS_main.py
Our model achieves the following performance on CIFAR-10, CIFAR100, CINIC10 and Fashion-MNIST:
Dataset | ||
---|---|---|
CIFAR10 | ||
CINIC10 | ||
Fashion-MNIST |