Create ann_train.cu