MANAGING TRAINING OF A MACHINE LEARNING MODEL

There is provided a method performed by a master node for managing training of a machine learning model. One or more worker nodes of a plurality of worker nodes are selected (102) to train a machine learning model in a round of training. The one or more worker nodes are selected to optimise a performance of an updated machine learning model for a validation dataset after the round of training. The updated machine learning model has one or more parameters of the machine learning model trained by the one or more worker nodes in a previous round of training. L'invention concerne un procédé réalisé par un nœud maître pour gérer l'entraînement d'un 5 modèle d'apprentissage automatique. Un ou plusieurs nœuds de travail d'une pluralité de nœuds de travail sont sélectionnés (102) pour entraîner un modèle d'apprentissage automatique dans un cycle d'entraînement. Le ou les nœuds de travail sont sélectionnés pour optimiser une performance d'un modèle d'apprentissage automatique mis à jour pour un ensemble de données de validation après le cycle d'entraînement. Le modèle d'apprentissage automatique mis à jour a un ou plusieurs paramètres du modèle d'apprentissage automatique entraîné par le ou les nœuds de travail dans un cycle d'apprentissage précédent.