Support variable batch size in throughput callback#20236

Borda merged 5 commits intoLightning-AI:masterLightning-AI/pytorch-lightning:masterfrom

alex-hh:throughput-callback-variable-batch-sizealex-hh/pytorch-lightning:throughput-callback-variable-batch-sizeCopy head branch name to clipboard