Hi:
I also want to implement the EWC in pytorch by myself.
But according to the Counterpart Tensorflow implementation:
https://github.com/ariseff/overcoming-catastrophic/blob/master/experiment.ipynb
their accuracy is so high.
Why does the Pytorch implementation has only 0.7 in your implementation.
Many thanks
Shixian Wen

@moskomule

Think https://github.com/moskomule/ewc.pytorch/blob/master/utils.py#L41 is the right way or not.

Hello, I have read your code and it is really helpful for me to understand the logic of EWC.
I still have a doubt for the input of nll_loss in _diag_fisher function. Should the label be the ground truth, not the output of the model?
Thanks for your code again and looking forward to your answer!

Hi,moskomule! Great work!
I want to know what version of pytorch did you use? Because I can't reproduce your results. I am wondering if the problem appeared on the version of pytorch.