pytorch-lightning

Support variable batch size in throughput callback

#20236

Merged