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…

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Sascha Heyer

Sascha Heyer

Hi, I am Sascha, Senior Machine Learning Engineer at @DoiT International and Founder of IOAnnotator an AI Platform.