Aperçu des expériences - CoreWeave Docs

Ces classes constituent les éléments de base pour suivre des expériences de machine learning, gérer des artifacts et configurer le comportement du SDK. Elles vous permettent de journaliser des métriques, de stocker des points de contrôle de modèles, de versionner des datasets et de gérer les configurations d’expériences, avec des fonctionnalités complètes de reproductibilité et de collaboration.

Pour en savoir plus sur l’utilisation de ces classes dans vos expériences de ML, consultez la documentation Experiments et Artifacts.

Classes principales

Classe	Description
`Run`	L’unité de calcul principale journalisée par W&B, qui représente une expérience de ML unique avec ses métriques, ses configurations et ses sorties.
`Artifact`	Élément de base souple et léger pour la gestion des versions des datasets et des modèles, avec déduplication automatique et suivi de la traçabilité.
`Settings`	Gestion de la configuration du SDK W&B, qui en contrôle le comportement, de la journalisation aux interactions avec l’API.

Premiers pas

Suivre une expérience

Créez et suivez une expérience de machine learning en journalisant ses métriques :

import wandb

# Initialiser un nouveau run
with wandb.init(project="my-experiments", config={"learning_rate": 0.001}) as run:
    # Accéder à la configuration
    config = run.config
    
    # Journaliser les métriques pendant l’entraînement
    for epoch in range(10):
        metrics = train_one_epoch()  # Votre logique d’entraînement
        run.log({
            "loss": metrics["loss"],
            "accuracy": metrics["accuracy"],
            "epoch": epoch
        })
    
    # Journaliser les métriques de synthèse
    run.summary["best_accuracy"] = max_accuracy

Versionner un artifact de modèle

Créez et journalisez un artifact de modèle versionné avec des métadonnées :

import wandb

with wandb.init(project="my-models") as run:
    # Entraîner votre modèle
    model = train_model()
    
    # Créer un artifact pour le modèle
    model_artifact = wandb.Artifact(
        name="my-model",
        type="model",
        description="ResNet-50 trained on ImageNet subset",
        metadata={
            "architecture": "ResNet-50",
            "dataset": "ImageNet-1K",
            "accuracy": 0.95
        }
    )
    
    # Ajouter les fichiers du modèle à l’artifact
    model_artifact.add_file("model.pt")
    model_artifact.add_dir("model_configs/")
    
    # Journaliser l’artifact dans W&B
    run.log_artifact(model_artifact)

Configurer les paramètres du SDK

Personnalisez le comportement du SDK W&B en fonction de vos exigences :

import wandb

# Configurer les paramètres par programmation
wandb.Settings(
    project="production-runs",
    entity="my-team",
    mode="offline",  # Exécuter hors ligne et synchroniser ultérieurement
    save_code=True,   # Enregistrer le code source
    quiet=True        # Réduire les sorties dans la console
)

# Ou utiliser des variables d’environnement
# export WANDB_PROJECT=production-runs
# export WANDB_MODE=offline

# Initialiser avec des paramètres personnalisés
with wandb.init() as run:
    # Placez ici le code de votre expérience
    pass

Lier des artifacts pour le suivi de la traçabilité

Suivez les relations entre les datasets, les modèles et les évaluations :

import wandb

with wandb.init(project="ml-pipeline") as run:
    # Utiliser un artifact de jeu de données
    dataset = run.use_artifact("dataset:v1")
    dataset_dir = dataset.download()
    
    # Entraîner le modèle sur le jeu de données
    model = train_on_dataset(dataset_dir)
    
    # Créer un artifact de modèle rattaché au jeu de données (traçabilité)
    model_artifact = wandb.Artifact(
        name="trained-model",
        type="model"
    )
    model_artifact.add_file("model.pt")
    
    # Journaliser avec suivi automatique de la traçabilité
    run.log_artifact(model_artifact)

​Classes principales

​Premiers pas

​Suivre une expérience

​Versionner un artifact de modèle

​Configurer les paramètres du SDK

​Lier des artifacts pour le suivi de la traçabilité