Passer au contenu principal
Une configuration de balayage se compose de paires clé-valeur imbriquées. Utilisez les clés de premier niveau de votre configuration de balayage pour définir les caractéristiques de votre recherche, comme les paramètres à explorer (clé parameter), la méthode de recherche dans l’espace des paramètres (clé method), entre autres. Le tableau suivant répertorie les clés de configuration de balayage de premier niveau, avec une brève description. Voir les sections correspondantes pour plus d’informations sur chaque clé. Voir la structure de Sweep configuration pour plus d’informations sur l’organisation de votre configuration de balayage.

metric

Utilisez la clé de configuration de niveau supérieur metric du balayage pour spécifier le nom, l’objectif et la métrique cible à optimiser.

parameters

Dans votre fichier YAML ou votre script Python, spécifiez parameters comme clé de premier niveau. Dans la clé parameters, indiquez le nom d’un hyperparamètre que vous souhaitez optimiser. Les hyperparamètres courants incluent : le taux d’apprentissage, la taille de lot, les époques, les optimiseurs, etc. Pour chaque hyperparamètre que vous définissez dans votre configuration de balayage, spécifiez une ou plusieurs contraintes de recherche. Le tableau suivant présente les contraintes de recherche d’hyperparamètres prises en charge. En fonction de votre hyperparamètre et de votre cas d’utilisation, utilisez l’une des contraintes de recherche ci-dessous pour indiquer à votre agent de balayage où rechercher (dans le cas d’une distribution) ou quoi rechercher/utiliser (value, values, etc.).
W&B définit les distributions suivantes selon les conditions ci-dessous si aucune distribution n’est spécifiée :
  • categorical si vous spécifiez values
  • int_uniform si vous spécifiez max et min comme entiers
  • uniform si vous spécifiez max et min comme nombres à virgule flottante
  • constant si vous affectez un ensemble à value

method

Spécifiez la stratégie de recherche d’hyperparamètres à l’aide de la clé method. Vous avez le choix entre trois stratégies de recherche d’hyperparamètres : recherche sur grille, aléatoire et recherche bayésienne. Parcourez chaque combinaison possible de valeurs d’hyperparamètre. La recherche sur grille choisit, à chaque itération, parmi un ensemble prédéfini de valeurs d’hyperparamètre, sans information préalable. La recherche sur grille peut être coûteuse en ressources de calcul. La recherche sur grille s’exécute indéfiniment si elle explore un espace de recherche continu. Choisissez, à chaque itération, un ensemble aléatoire de valeurs d’hyperparamètres, sans information préalable, selon une distribution. La recherche aléatoire s’exécute indéfiniment, sauf si vous arrêtez le processus depuis la ligne de commande, dans votre script Python ou dans application W&B. Spécifiez l’espace de distribution avec la clé de métrique si vous choisissez la recherche aléatoire (method: random). Contrairement à la recherche aléatoire et à la recherche sur grille, les modèles bayésiens prennent des décisions éclairées. L’optimisation bayésienne s’appuie sur un modèle probabiliste pour déterminer quelles valeurs utiliser, via un processus itératif qui teste des valeurs sur une fonction de substitution avant d’évaluer la fonction objectif. La recherche bayésienne est efficace pour un petit nombre de paramètres continus, mais elle passe mal à l’échelle. Pour en savoir plus sur la recherche bayésienne, voir le Bayesian Optimization Primer. La recherche bayésienne se poursuit indéfiniment, à moins que vous n’arrêtiez le processus depuis la ligne de commande, dans votre script Python ou dans l’application W&B. Dans la clé parameter, imbriquez le nom de l’hyperparamètre. Ensuite, spécifiez la clé distribution et indiquez une distribution pour la valeur. Le tableau suivant répertorie les distributions prises en charge par W&B.

early_terminate

Utilisez l’arrêt anticipé (early_terminate) pour interrompre les runs peu performants. En cas d’arrêt anticipé, W&B arrête le run en cours avant d’en créer un nouveau avec un nouvel ensemble de valeurs d’hyperparamètres.
Vous devez spécifier un algorithme d’arrêt si vous utilisez early_terminate. Imbriquez la clé type dans early_terminate au sein de votre configuration de balayage.

Algorithme d’arrêt

W&B prend actuellement en charge l’algorithme d’arrêt Hyperband.
L’optimisation des hyperparamètres avec Hyperband détermine si un programme doit s’arrêter ou continuer à une ou plusieurs itérations prédéfinies, appelées brackets. Lorsqu’un run W&B atteint un bracket, le balayage compare la métrique de ce run à toutes les valeurs de métrique précédemment enregistrées. Le balayage met fin au run si la valeur de sa métrique est trop élevée (lorsque l’objectif est la minimisation) ou si la métrique du run est trop faible (lorsque l’objectif est la maximisation). Les brackets sont basés sur le nombre d’itérations enregistrées. Le nombre de brackets correspond au nombre de fois où vous enregistrez la métrique que vous optimisez. Les itérations peuvent correspondre à des étapes, des époques ou à quelque chose entre les deux. La valeur numérique du compteur d’étapes n’est pas utilisée dans les calculs des brackets.
Spécifiez min_iter ou max_iter pour créer une planification de brackets.
Hyperband vérifie quels runs arrêter toutes les quelques minutes. L’horodatage de fin du run peut différer des brackets spécifiés si votre run est court ou si vos itérations sont courtes.

command

Modifiez le format et le contenu à l’aide de valeurs imbriquées dans la clé command. Vous pouvez inclure directement des composants fixes, comme des noms de fichiers.
Sur les systèmes Unix, /usr/bin/env garantit que le système d’exploitation choisit le bon interpréteur Python en fonction de l’environnement.
W&B prend en charge les macros suivantes pour les composants variables de la commande :