Lightning-AI/pytorch-lightning

An example for knowledge distillation. Especially for the load the teacher model's weights , and train the student model

Open

#17.592 aberto em 8 de mai. de 2023

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

Métricas do repositório

Stars
 (26.687 stars)
Métricas de merge de PR
 (Mesclagem média 9d 15h) (3 fundiu PRs em 30d)

Description

Description & Motivation

No response

Pitch

A example for knowledge distillation. Especially for the load the teacher model's weight ,and train the student model.

Now I have a trained teacher_model. When I want to train the student model, I need to load the teacher model's weights and set the teacher model to eval().

I am confused about the specific steps to do above using the Lightning. Could I get an example for it?

Alternatives

An example for knowledge distillation. Especially for the load the teacher model's weight ,and train the student model

Additional context

An example for knowledge distillation. Especially for the load the teacher model's weight ,and train the student model.

cc @borda

Guia do colaborador

Direção de pesquisa
Procure exemplos existentes de destilação de conhecimento no repositório Lightning ou forneça orientação sobre como instanciar um modelo professor, carregar seus pesos pré treinados, configurá lo no modo de avaliação e definir um modelo aluno. Implemente um LightningModule que lide com a perda de destilação e a lógica de treinamento.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Documentação
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
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.
70