This repo is associated with the paper Simulated Annealing in Early Layers Leads to Better Generalization (CVPR 2023).
python3 -m venv venv
source venv/bin/activate
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt
python main.py --set tinyImagenet_full --data_dir=<u>DataRoot</u> --arch=ResNet50 --exp_mode=ascend
Refer to config.py if you wish to try different hyper parameters/settings.