To install requirements:
pip install -r requirements.txt
To train the model(s) in the paper, run this command:
python3 main.py --data Cifar10 --acqMode Max_Diversity --optimizer_name Adam --isInit True
python3 main.py --data Cifar10 --acqMode Max_Diversity --optimizer_name SAM --isInit False
- Data will be downloaded to folder 'data'.
- Result will be recorded to folder 'Results'.
Our model achieves the following performance on active learning settings:
Optimizer | FashionMNIST | SVHN | CIFAR-10 | CIFAR-100 |
---|---|---|---|---|
Adam | 85.8% | 76.8% | 54.4% | 47.6% |
SAM | 86.3% | 78.8% | 57.0% | 48.4% |