Lightning-AI/pytorch-lightning

Make logging_mode (`on_step` or `on_epoch`) available to loggers

Open

#20.502 aperta il 17 dic 2024

Vedi su GitHub
 (5 commenti) (0 reazioni) (0 assegnatari)Python (3233 fork)batch import
featurehelp wanted

Metriche repository

Star
 (26.687 star)
Metriche merge PR
 (Merge medio 9g 15h) (3 PR mergiate in 30 g)

Descrizione

Description & Motivation

Lightning lets you log on_step, or on_epoch. However, even when using on_epoch, the global step is logged on the x-axis, not the epoch number.

Loggers can manually extract epoch from the logged metrics and pass that as the step. However, this would break on_step logging.

Making the logging mode for a logging call available would let loggers conditionally use either the global_step or current_epoch as the step for that particular metric.

Pitch

No response

Alternatives

No response

Additional context

No response

cc @lantiga @borda

Guida contributor