Skip to content

aosong01/MoASE-Pytorch

 
 

Repository files navigation

Decomposing the Neurons: Activation Sparsity via Mixture of Experts for Continual Test Time Adaptation

Installation

Please create and activate the following conda envrionment.

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

Classification Experiments

  • ViT as the backbone

Our source model is from timm, you can directly donwload it from the code.

Cifar10-to-Cifar10C task

Please load the source model from here

bash run_cifar10.sh # MoASE

Cifar100-to-Cifar100C task

Please load the source model from here

cd cifar
bash run_cifar100.sh # MoASE

For segmentation code, you can refer to cotta and SVDP. As for the source model, you can directly use Segformer trained on Cityscapes.

Citation

Please cite our work if you find it useful.

Acknowledgement

Data links

About

[Official code for MoASE]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.7%
  • Other 0.3%