Initialiser un balayage

Initialisez un balayage W&B à l’aide du SDK Python ou de la CLI pour lancer des recherches d’hyperparamètres avec votre configuration de balayage.

W&B utilise un contrôleur de balayage pour gérer les balayages dans le cloud (standard) ou localement (local), sur une ou plusieurs machines. Une fois qu’un run est terminé, le contrôleur de balayage émet un nouvel ensemble d’instructions décrivant un nouveau run à exécuter. Ces instructions sont prises en charge par des agents qui exécutent effectivement les runs. Dans un balayage W&B classique, le contrôleur s’exécute sur le serveur W&B. Les agents s’exécutent sur vos machines.

Les extraits de code suivants montrent comment initialiser des balayages avec la CLI et dans un notebook Jupyter ou un script Python.

1. Avant d’initialiser un balayage, assurez-vous d’avoir défini une configuration de balayage soit dans un fichier YAML, soit dans un objet dictionnaire Python imbriqué dans votre script. Pour plus d’informations, voir Définir une configuration de balayage.
2. Le balayage W&B et le run W&B doivent tous deux se trouver dans le même projet. Par conséquent, le nom que vous fournissez lorsque vous initialisez W&B (wandb.init()) doit correspondre au nom du projet que vous fournissez lorsque vous initialisez un balayage W&B (wandb.sweep()).

Script Python ou notebook

Utilisez le SDK W&B pour initialiser un balayage. Passez le dictionnaire de configuration du balayage au paramètre sweep. Vous pouvez également fournir le nom du projet via le paramètre de projet (project) dans lequel vous souhaitez stocker la sortie du run W&B. Si le projet n’est pas spécifié, le run est placé dans un projet « Non catégorisé ».

import wandb

# Exemple de configuration de balayage
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 du balayage. L’ID du balayage inclut le nom de l’entité et le nom du projet. Notez l’ID du balayage.

CLI

Utilisez la CLI W&B pour initialiser un balayage. Indiquez le nom de votre fichier de configuration. Vous pouvez également fournir le nom du projet avec l’indicateur project. Si le projet n’est pas spécifié, le run W&B est placé dans un projet « Non catégorisé ».

Utilisez la commande wandb sweep pour initialiser un balayage. L’exemple de code suivant initialise un balayage pour un projet sweeps_demo et utilise un fichier config.yaml comme configuration.

wandb sweep --project sweeps_demo config.yaml

Cette commande affiche un ID de balayage. L’ID du balayage inclut le nom de l’entité et le nom du projet. Notez l’ID du balayage.