- Install PyTorch.
- Install required python packages using the following command:
pip install -r requirements.txt
- Training and Evaluation data:
- CIFAR10, CIFAR100, and Tiny ImageNet
Run the following command:
CUDA_VISIBLE_DEVICES=0 python TADAT.py --eta 4.0 --lamda 0.4Run the following command:
CUDA_VISIBLE_DEVICES=0 python eval_AA.py --arch '' --dataset '' --model_path '' --model_name '' --save_file_name ''The code refers to BSL-AT.
We thank the authors for sharing sincerely.
This project is released under the MIT License.