Lightning-AI/pytorch-lightning

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

Open

#20,502 opened on 2024年12月17日

GitHub で見る
 (5 comments) (0 reactions) (0 assignees)Python (26,687 stars) (3,233 forks)batch import
featurehelp wanted

説明

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

コントリビューターガイド

技術スタック
pythonpytorch
領域
machine learningdeveloper experience
Issue 種別
feature
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
3
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
half day
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
fresh
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
clear
前提条件
PythonPyTorchLightning logging API
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
60
調査方針
Investigate the Logger base class in the Lightning codebase, specifically how metrics are passed to loggers (e.g., the `log` method). Look at the `LightningLoggerBase` class and how `global step` and `current epoch` are currently used. Determine where the logging mode is available and how it can be exposed to loggers. Review any related issues or comments in this thread for additional context.