Passer au contenu principal
Après avoir défini une configuration de sweep, et avant de lancer des agents, initialisez un sweep pour enregistrer sa configuration dans W&B et recevoir un ID de sweep. Les agents utilisent l’ID de sweep pour récupérer les instructions d’exécution. Une fois un sweep initialisé, il est prêt à coordonner des recherches d’hyperparamètres sur une ou plusieurs machines. W&B utilise un contrôleur de sweep pour gérer les sweeps sur une ou plusieurs machines. Une fois qu’un run est terminé, le contrôleur génère un nouvel ensemble d’instructions décrivant le run suivant à exécuter, puis les agents récupèrent ces instructions et les exécutent. Par défaut, W&B héberge le contrôleur, et les agents s’exécutent sur vos machines. Avant d’initialiser un sweep, vous devez avoir défini une configuration de sweep soit dans un fichier YAML, soit dans un dictionnaire Python imbriqué dans votre script. Pour plus d’informations, voir Define sweep configuration. Les extraits de code suivants montrent comment initialiser des sweeps avec la CLI et dans un notebook Jupyter ou un script Python. Choisissez la méthode qui correspond à votre flux de travail.
Le sweep et le run doivent se trouver dans le même projet. Le nom du projet que vous fournissez lorsque vous initialisez W&B (wandb.init()) doit correspondre au nom du projet que vous fournissez lorsque vous initialisez un sweep (wandb.sweep()).
Utilisez le SDK W&B pour initialiser un sweep. Passez le dictionnaire de configuration du sweep au paramètre sweep. Vous pouvez éventuellement définir le paramètre project avec le nom du projet dans lequel vous souhaitez que W&B stocke la sortie du run. Si vous ne spécifiez pas de projet, W&B place le run dans un projet Uncategorized.
La fonction wandb.sweep() renvoie l’ID du sweep. L’ID du sweep inclut le nom de l’entité et le nom du projet. Notez l’ID du sweep : vous le transmettrez aux agents lorsque vous les démarrerez. Avec l’ID du sweep, vous êtes prêt à lancer un ou plusieurs agents pour exécuter des runs.