How do we train this on gpu?
How do we train this on gpu?