Speed up training with GPUs