To run the cifar-10 experiment:
cd cifar10
python run_TMNs_cifar10.py
To run the ImageNet-50 experiment:
cd imagenet-50
python run_TMNs_imagenet.py
Please cite our paper if it is helpful to your work:
@article{wang2021triple,
title={Triple-memory networks: A brain-inspired method for continual learning},
author={Wang, Liyuan and Lei, Bo and Li, Qian and Su, Hang and Zhu, Jun and Zhong, Yi},
journal={IEEE Transactions on Neural Networks and Learning Systems},
volume={33},
number={5},
pages={1925--1934},
year={2021},
publisher={IEEE}
}