Lightning-AI/pytorch-lightning

Provide a non-expert mode for the DeviceStatsMonitor

Open

#18.652 geöffnet am 27. Sept. 2023

Auf GitHub ansehen
 (0 Kommentare) (15 Reaktionen) (0 zugewiesene Personen)Python (3.233 Forks)batch import
callback: device statsfeaturehelp wanted

Repository-Metriken

Stars
 (26.687 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 9T 15h) (3 gemergte PRs in 30 T)

Beschreibung

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

Contributor Guide

Research-Richtung
Untersuchen Sie die aktuelle Protokollierung von torch.cuda.memory stats in DeviceStatsMonitor, entwerfen Sie ein Konfigurationsflag zur Auswahl zwischen Experten und vereinfachtem Logging und implementieren Sie dann die Filterung der Kernmetriken.
Tech Stack
pythonpytorch
Domain
backendperformanceobservability
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
2
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-3 Stunden
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
Python
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
75