Do you have a public repo for that? I'm also trying to do that although I'm using "replay"/distillation and hopfield memory banks.