what is crit argument in ewc = ElasticWeightConsolidation(model, crit, lr=0.01, weight=0.1)? and how to build it ?