nemo_rl.distributed.worker_group_utils

Contenu du Module

Fonctions

API

nemo_rl.distributed.worker_group_utils.get_nsight_config_if_pattern_matches(worker_name: str) -> dict[str, typing.Any]

Vérifier si le nom du worker correspond aux motifs dans NRL_NSYS_WORKER_PATTERNS et renvoyer la configuration nsight.

Arguments : worker_name : Nom du worker à vérifier par rapport aux motifs

Renvoie : Dictionnaire contenant {“nsight”: config} si le motif correspond, dictionnaire vide sinon

nemo_rl.distributed.worker_group_utils.recursive_merge_options(
    default_options: dict[str, typing.Any], extra_options: dict[str, typing.Any]
) -> dict[str, typing.Any]

Fusionner récursivement les options supplémentaires dans les options par défaut à l’aide d’OmegaConf.

Arguments : default_options : Dictionnaire des options par défaut (priorité inférieure) extra_options : Options supplémentaires fournies par l’appelant (priorité supérieure)

Renvoie : Dictionnaire des options fusionnées avec extra_options ayant la priorité sur default_options