This is the official code of the work Enhanced Online Test-time Adaptation with Feature-Weight Cosine Alignment (2024)
- Python 3.8
- PyTorch 1.9.0
Download datasets from here or using the links below:
Run bash script/<dataset>.sh
for training.
Example:
bash script cifar100.sh
Or you can manually configure the training parameter by:
python -u main.py --model_path <path_to_checkpoint> \
--data_path <path_to_dataset> \
--source_dataset cifar-100 \
--target_dataset cifar-100-c \
--lr 0.005 \
--tta_batchsize 128 \
--severity 5 \
--criterion entropy \
--network wrn-40x2