Lightning-AI/pytorch-lightning
Auf GitHub ansehenMake logging_mode (`on_step` or `on_epoch`) available to loggers
Open
#20.502 geöffnet am 17. Dez. 2024
featurehelp wanted
Repository-Metriken
- Stars
- (26.687 Stars)
- PR-Merge-Metriken
- (Durchschn. Merge 9T 15h) (3 gemergte PRs in 30 T)
Beschreibung
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