nemo_rl

Sous-packages

• experience
• metrics
• evals
• distributed
• environments
• algorithms
• converters
• utils
• models
• data

Sous-modules

• package_info

Contenu du module

Fonctions

Données

megatron_path

API

nemo_rl.megatron_path

Valeur: None

nemo_rl._patch_nsight_file()

Corriger le fichier nsight.py pour résoudre l’affectation de context.py_executable.

Jusqu’à ce que cette correction soit intégrée en amont, nous maintiendrons ce correctif ici. Cette logique de correction n’est appliquée que si l’utilisateur souhaite utiliser le profilage nsys, qu’il active avec NRL_NSYS_WORKER_PATTERNS.

Si activé, appliquera efficacement le correctif suivant de manière idempotente :

https://github.com/ray-project/ray/compare/master…terrykong:ray:tk/nsight-py-exeutable-fix?expand=1

Ce hack fonctionne car le plugin nsight n’est pas appelé à partir du processus pilote principal, donc dès que nemo_rl est importé, le correctif est appliqué et la source du module nsight.py est à jour avant que nsight.py ne soit réellement nécessaire.