Passer à la navigation

nemo_rl.utils.native_checkpoint

Afficher en Markdown

Utilitaires de gestion des points de contrôle pour les modèles HF.

Contenu du Module

Classes

NomDescription
ModelStateClasse d’assistance pour suivre l’état du modèle dans les points de contrôle distribués.
OptimizerStateClasse d’assistance pour suivre l’état de l’optimiseur dans les points de contrôle distribués.

Fonctions

NomDescription
save_checkpointEnregistrer un point de contrôle du modèle et éventuellement l’état de l’optimiseur.
load_checkpointCharger les poids du modèle et éventuellement l’état de l’optimiseur.
convert_dcp_to_hfConvertir un point de contrôle Torch DCP en un point de contrôle Hugging Face.

API

(Rest of the document remains the same as the original, only translating text nodes)

The rest of the technical MDX document remains structurally identical, with only text content translated to French.