Skip to content
/ EWC Public

TensorFlow implementation of Elastic Weight Consolidation

Notifications You must be signed in to change notification settings

stokesj/EWC

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Elastic weight consolidation

Introduction

A TensorFlow implementation of elastic weight consolidation as presented in Overcoming catastrophic forgetting in neural networks.

Usage

Perform hyperparameter search over learning rates for the permuted MNIST task (fisher multiplier locked at inverse learning rate):

python -u main.py --hidden_layers 2 --hidden_units 800 --num_perms 5 --trials 50 --epochs 100

Results



About

TensorFlow implementation of Elastic Weight Consolidation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages