The original EWC requires you to compute the importance for each weight based on an additional pass over the training set. The importance is the squared gradient averaged over each minibatch. Anyway, you can take a look at the implementation available in the ContinualAI notebooks 147. It is an association for Continual Learning 
Disclaimer: I am part of ContinualAI.