You have developed a machine learning model using scikit-learn, but you are experiencing longer than expected training times. To address this issue, you decide to migrate your model to Vertex AI Training offered by Google Cloud. Your goal is to improve the training time of the model. Based on this scenario, what should you try first to achieve your goal?