IL2A icon indicating copy to clipboard operation
IL2A copied to clipboard

PyTorch implementation of our NeurIPS 2021 paper "Class-Incremental Learning via Dual Augmentation"

IL2A - Official PyTorch Implementation

[NeurIPS2021] Class-Incremental Learning via Dual Augmentation

Fei Zhu, Zhen Cheng, Xu-Yao Zhang, Cheng-Lin Liu
Paper

Usage

We run the code with torch version: 1.10.0, python version: 3.9.7

  • Train CIFAR100
cd cifar
python main.py
  • Train Tiny-ImageNet
cd tiny
python main_tiny.py

Citation

@article{zhu2021class,
  title={Class-Incremental Learning via Dual Augmentation},
  author={Zhu, Fei and Cheng, Zhen and Zhang, Xu-Yao and Liu, Cheng-lin},
  journal={Advances in Neural Information Processing Systems},
  volume={34},
  pages={14306--14318},
  year={2021}
}

Contact

Fei Zhu ([email protected])