Update pre_train_objects.cu