Official code for Controllable Continual Test-Time Adaptation. Here, we use the CIFAR10-C dataset as an example to introduce the usage of the code. [arxiv]

Please create and activate the following conda envrionment. To reproduce our results, please kindly create and use this environment.

# It may take several minutes for conda to solve the environment
conda update conda
conda env create -f environment.yml
conda activate ccotta 

Please download the dataset CIFAR10-C to any path by yourself, and fill in the path in _C.DATA_DIR in conf.py.

python test_time.py --cfg ./cfgs/cifar10_c/CCoTTA.yaml --gpu 0 --cav_alpha 2.0 --cav_beta 0.05

If you find our survey and repository useful for your research, please consider citing our paper:

@article{shi2024controllable,
  title={Controllable Continual Test-Time Adaptation},
  author={Shi, Ziqi and Lyu, Fan and Liu, Ye and Shang, Fanhua and Hu, Fuyuan and Feng, Wei and Zhang, Zhang and Wang, Liang},
  journal={arXiv preprint arXiv:2405.14602},
  year={2024}
}