Lightning-AI/pytorch-lightning

Provide a non-expert mode for the DeviceStatsMonitor

Open

#18.652 aberto em 27 de set. de 2023

Ver no GitHub
 (0 comments) (15 reactions) (0 assignees)Python (3.233 forks)batch import
callback: device statsfeaturehelp 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

The DeviceStatsMonitor currently dumps the entire torch.cuda.memory_stats into the logger. Beginner users might be overwhelmed by the amount of data logged and are confused which metrics matter to maximize their utilization.

Pitch

Make the callback more user-friendly by logging only core metrics like GPU utilization and total memory usage. Provide an expert mode that when enabled logs what we see today.

Alternatives

No response

Additional context

This was proposed by user magical_arjay on our Discord.

cc @borda @awaelchli

Guia do colaborador

Direção de pesquisa
Examine o registro atual de torch.cuda.memory stats no DeviceStatsMonitor, projete uma flag de configuração para selecionar entre logging especialista e simplificado e, em seguida, implemente a filtragem das métricas principais.
Pilha de tecnologia
pythonpytorch
Domain
backendperformanceobservability
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.
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
Python
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
75