Skip to content

Wandb 1.x step handling #20368

@edmcman

Description

@edmcman

Description & Motivation

I recently just upgraded from lightning 1.x to 2.x. During the change, it seems that lightning has changed the way that steps are handled when logging to wandb. In particular, instead of using wandb's step argument, as was done in 1.x, lightning 2.x's step is logged as trainer/global_step. The trouble for me is that this makes comparing runs between 1.x and 2.x very difficult, because the steps are not on the same scale.

Pitch

I would like an option to WandbLogger that allows me to optionally configure the 1.x step behavior.

Alternatives

No response

Additional context

image
image

cc @lantiga @Borda @morganmcg1 @borisdayma @scottire @parambharat

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureIs an improvement or enhancementlogger: wandbWeights & Biases

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions