/ewc.pytorch

An implementation of EWC with PyTorch

Primary LanguageJupyter Notebook

EWC.pytorch

An implementation of Elastic Weight Consolidation (EWC), proposed in James Kirkpatrick et al. Overcoming catastrophic forgetting in neural networks 2016(10.1073/pnas.1611835114).

  • demo.ipynb demonstrates EWC with supervised learning.