Update pre_train_faces(auto).cu