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

cc @Borda