nemo_rl.models.policy.megatron_policy_worker

Contenu du module

Nom	Description
`MegatronPolicyWorker`	Aucun

Nom	Description
`broadcast_object_across_pp_ranks`	Diffuser un objet à travers les rangs parallèles du pipeline.
`setup_megatron_model`	Aucun
`destroy_parallel_state`	Détruire de manière sécurisée l’état parallèle et réinitialiser le suivi des appels asynchrones.

TokenizerType

(Reste du document traduit de la même manière, en conservant tous les blocs de code et la structure MDX, en ne traduisant que les textes visibles)