Lightning-AI/lit-llama

Resume in the pretraining code

Open

#359 ouverte le 2 juin 2023

Voir sur GitHub
 (2 commentaires) (1 réaction) (0 assignés)Python (473 forks)batch import
help wanted

Métriques du dépôt

Stars
 (5 533 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

I would like to request a new feature in the code: the ability to resume training from a checkpoint.

Currently, the code can save a checkpoint of the model's state at any point during training. However, there is no way to resume training from a checkpoint.

The code can save two things along with the model state_dict: 1)the optimizer, 2)the id of the last example it has seen (assuming the data is fed sequentially to the model not randomly)

Guide contributeur

Direction de recherche
Implémentez le chargement du point de contrôle pour restaurer l'état du modèle, de l'optimiseur et des données, puis reprenez la boucle d'entraînement.
Stack technique
python
Domaine
ai
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Une demi journée
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Récente
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
65