Show actual steps instead of the length of dataset on the progress bar

[Boltzmachine]

Description & Motivation

When we set max_steps, the progress bar still show the length of the dataset. This is inconvenient as one needs to calculate the remaining time themself. Showing the length of dataset is actually not usable, especiall when the number of steps within an epoch is larger than max_steps.

Refer to #20124

Pitch

No response

Alternatives

No response

Additional context

No response

cc @lantiga @Borda