Integrate total_batch_idx
with progress tracking
#8598
Merged
total_batch_idx
with progress tracking
#8598