Passer au contenu principal

Voir le code source

Ajoute une expérience Launch W&B à la file d’attente, à partir d’un URI source, d’un job ou de docker_image.
launch_add(
    uri: Optional[str] = None,
    job: Optional[str] = None,
    config: Optional[Dict[str, Any]] = None,
    template_variables: Optional[Dict[str, Union[float, int, str]]] = None,
    project: Optional[str] = None,
    entity: Optional[str] = None,
    queue_name: Optional[str] = None,
    resource: Optional[str] = None,
    entry_point: Optional[List[str]] = None,
    name: Optional[str] = None,
    version: Optional[str] = None,
    docker_image: Optional[str] = None,
    project_queue: Optional[str] = None,
    resource_args: Optional[Dict[str, Any]] = None,
    run_id: Optional[str] = None,
    build: Optional[bool] = (False),
    repository: Optional[str] = None,
    sweep_id: Optional[str] = None,
    author: Optional[str] = None,
    priority: Optional[int] = None
) -> "public.QueuedRun"
ArgumentsDescription
uriURI de l’expérience à exécuter. URI d’un run wandb ou URI d’un dépôt Git.
jobréférence de chaîne vers un wandb.Job, par ex. : wandb/test/my-job:latest
configUn dictionnaire contenant la configuration du run. Peut également contenir des arguments spécifiques aux ressources sous la clé “resource_args”
template_variablesUn dictionnaire contenant les valeurs des variables de modèle pour une file d’attente de runs. Format attendu : {"VAR_NAME": VAR_VALUE}
projectprojet cible vers lequel envoyer le run lancé
entityentité cible vers laquelle envoyer le run lancé
queuele nom de la file d’attente dans laquelle placer le run
priorityle niveau de priorité du job, où 1 est la priorité la plus élevée
resourcebackend d’exécution du run : W&B prend en charge nativement le backend “local-container”
entry_pointPoint d’entrée à exécuter dans le projet. Par défaut, utilise le point d’entrée du run original pour les URI wandb, ou main.py pour les URI de dépôt Git.
nameNom sous lequel lancer le run.
versionPour les projets basés sur Git, soit un hash de commit, soit un nom de branche.
docker_imageLe nom de l’image Docker à utiliser pour le run.
resource_argsArguments liés aux ressources pour lancer des runs sur un backend distant. Seront stockés dans la configuration Launch générée sous resource_args.
run_idchaîne facultative indiquant l’ID du run lancé
buildindicateur facultatif dont la valeur par défaut est false ; si build est défini, queue doit aussi l’être ; une image est créée, un artifact de job est créé, puis une référence à cet artifact de job est envoyée à queue
repositorychaîne facultative permettant de contrôler le nom du dépôt distant, utilisée lors de l’envoi d’images vers un registre
project_queuechaîne facultative permettant de contrôler le nom du projet pour la file d’attente. Principalement utilisée pour la rétrocompatibilité avec les files d’attente limitées au projet

Exemple :

from wandb.sdk.launch import launch_add

project_uri = "https://github.com/wandb/examples"
params = {"alpha": 0.5, "l1_ratio": 0.01}
# Exécuter le projet W&B et créer un environnement docker reproductible
# sur un hôte local
api = wandb.apis.internal.Api()
launch_add(uri=project_uri, parameters=params)
RenvoieDescription
une instance de wandb.api.public.QueuedRun qui fournit des informations sur le run en file d’attente ou, si wait_until_started ou wait_until_finished sont appelés, donne accès aux informations du run sous-jacent.
Exceptions levéesDescription
wandb.exceptions.LaunchError en cas d’échec