Lightning-AI/pytorch-lightning
Ver no GitHubMake logging_mode (`on_step` or `on_epoch`) available to loggers
Open
#20.502 aberto em 17 de dez. de 2024
featurehelp 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
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