fastprogress_nested_progress_bar_training_loop_example.py

1from fastprogress import master_bar, progress_bar
2import time
3
4# Total number of epochs and batches
5epochs = 5
6batches_per_epoch = 10
7
8# Create the master progress bar for the outer loop (epochs)
9mb = master_bar(range(epochs))
10
11for epoch in mb:
12    # Create the child progress bar for the inner loop (batches)
13    # The parent=mb argument links it to the master bar
14    pb = progress_bar(range(batches_per_epoch), parent=mb)
15    
16    for batch in pb:
17        # Simulate work
18        time.sleep(0.1)
19        
20        # Optionally update the text on the master bar
21        mb.child.comment = f'Batch {batch}'
22    
23    # Update the master bar with a message after each epoch
24    mb.write(f'Finished epoch {epoch}')