output_size=10, # 10 classes - digits 0-9 ewc=True, # use EWC lam=args.lam # the lambda (fisher multiplier) value to be used in the EWC loss formula ).to(device) # a list of the models we instantiated above models = [sgd_dropout_model, ewc_model] models = setup.build_mode...