Lightning-AI/pytorch-lightning
Voir sur GitHubMake logging_mode (`on_step` or `on_epoch`) available to loggers
Open
#20 502 ouverte le 17 déc. 2024
featurehelp wanted
Métriques du dépôt
- Stars
- (26 687 stars)
- Métriques de merge PR
- (Merge moyen 9j 15h) (3 PRs mergées en 30 j)
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