Train Machine Learning models with Vertex AI Training
A simple and scalable approach with custom container
I regularly guide customers on how they can scale their training on the Cloud. While every customer has unique requirements and expectations, the overall process is the same.
In this article, we train a transformer model (DistilBert) with Vertex AI. As a machine learning Framework, I use TensorFlow. It will also work for any other machine learning framework (PyTorch, XGBoost, scikit-learn, you name it) and any other ML use…