Remove the 3rd class of train data. Add cuda kernel file and modify the makefile.