changes to get adversarial stuff on gpu