Lightning-AI/lit-llama

Resume in the pretraining code

Open

#359 aberto em 2 de jun. de 2023

Ver no GitHub
 (2 comments) (1 reaction) (0 assignees)Python (473 forks)batch import
help wanted

Métricas do repositório

Stars
 (5.533 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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)

Guia do colaborador

Direção de pesquisa
Implemente o carregamento do checkpoint para restaurar o estado do modelo, otimizador e dados, e então retome o loop de treinamento.
Pilha de tecnologia
python
Domain
ai
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Recente
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
65