Update pre_train_sample_creation.cu