Initialiser un sweep - CoreWeave Docs

Après avoir défini une configuration de sweep et avant de lancer des agents, initialisez un sweep afin d’enregistrer sa configuration auprès de W&B et d’obtenir un ID de sweep. Les agents se servent de cet ID pour récupérer les instructions d’exécution des runs. Une fois initialisé, le sweep 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. Chaque fois qu’un run se termine, le contrôleur émet un nouvel ensemble d’instructions décrivant le prochain run à exécuter, que les agents récupèrent puis exécutent. Par défaut, le contrôleur est hébergé par W&B 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 sous la forme d’un dictionnaire Python imbriqué dans votre script. Pour plus d’informations, voir Définir la configuration de sweep. Les extraits de code suivants montrent comment initialiser des sweeps avec la CLI, ainsi que dans un notebook Jupyter ou un script Python. Choisissez la méthode adaptée à votre flux de travail.

Le sweep et le run doivent appartenir au même projet. Le nom de projet que vous fournissez lors de l’initialisation de W&B (wandb.init()) doit correspondre à celui que vous fournissez lors de l’initialisation du sweep (wandb.sweep()).

Script Python ou notebook
CLI

Utilisez le SDK W&B pour initialiser un sweep. Transmettez le dictionnaire de configuration de sweep au paramètre sweep. Vous pouvez également affecter au paramètre project le nom du projet dans lequel W&B doit stocker la sortie du run. Si vous ne spécifiez pas de projet, W&B place le run dans un projet « Uncategorized ».

import wandb

# Exemple de configuration de sweep
sweep_configuration = {
    "method": "random",
    "name": "sweep",
    "metric": {"goal": "maximize", "name": "val_acc"},
    "parameters": {
        "batch_size": {"values": [16, 32, 64]},
        "epochs": {"values": [5, 10, 15]},
        "lr": {"max": 0.1, "min": 0.0001},
    },
}

sweep_id = wandb.sweep(sweep=sweep_configuration, project="project-name")

La fonction wandb.sweep() renvoie l’ID de sweep, qui inclut le nom de l’entity et le nom du projet. Notez cet ID : vous le transmettrez aux agents au moment de les démarrer. Une fois l’ID de sweep obtenu, vous pouvez lancer un ou plusieurs agents pour exécuter des runs.

Utilisez la CLI W&B pour initialiser un sweep. Indiquez le nom de votre fichier de configuration. Vous pouvez également affecter à l’option --project le nom du projet. Si vous ne spécifiez pas de projet, W&B place le run dans un projet « Uncategorized ».Utilisez la commande wandb sweep pour initialiser un sweep. L’exemple de code suivant initialise un sweep pour un projet sweeps_demo à partir d’un fichier de configuration config.yaml.

wandb sweep --project sweeps_demo config.yaml

Cette commande affiche un ID de sweep, qui inclut le nom de l’entity et le nom du projet. Notez cet ID : vous le transmettrez aux agents au moment de les démarrer. Une fois l’ID de sweep obtenu, vous pouvez lancer un ou plusieurs agents pour exécuter des runs.